fix args parser for train_ms (fishaudio#378)

fix the bug for fishaudio@81e034b
SynthAether · Jun 26, 2024 · 7d0aa66 · 7d0aa66
1 parent 669ac0e
commit 7d0aa66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_ms.py b/train_ms.py
@@ -99,7 +99,7 @@ def run():
         help="数据集文件夹路径，请注意，数据不再默认放在/logs文件夹下。如果需要用命令行配置，请声明相对于根目录的路径",
         default=config.dataset_path,
     )
-    args = parser.parse_known_args()
+    args, _ = parser.parse_known_args()
     model_dir = os.path.join(args.model, config.train_ms_config.model)
     if not os.path.exists(model_dir):
         os.makedirs(model_dir, exist_ok=True)