Merge pull request #20 from yyua8222/yy01071

update the args variable for checkpoint path

audioldm_train/infer.py

@@ -109,6 +109,13 @@ def infer(dataset_json, configs, config_yaml_path, exp_group_name, exp_name):
         required=False,
         help="The filelist that contain captions (and optionally filenames)",
     )
+    parser.add_argument(
+        "-reload_from_ckpt",
+        "--reload_from_ckpt",
+        type=str,
+        required=True,
+        help="the checkpoint path for the model",
+    )
 
     args = parser.parse_args()
 
@@ -122,7 +129,7 @@ def infer(dataset_json, configs, config_yaml_path, exp_group_name, exp_name):
     config_yaml_path = os.path.join(config_yaml)
     config_yaml = yaml.load(open(config_yaml_path, "r"), Loader=yaml.FullLoader)
 
-    if "reload_from_ckpt" is not None:
+    if args.reload_from_ckpt != None:
         config_yaml["reload_from_ckpt"] = args.reload_from_ckpt
 
     infer(dataset_json, config_yaml, config_yaml_path, exp_group_name, exp_name)