Fix the bug of mkdir in train_ms.py (#292)

Co-authored-by: 李盛强 <[email protected]>
fishaudio · Jan 18, 2024 · 90b67fe · 90b67fe
1 parent 8f4c1a5
commit 90b67fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_ms.py b/train_ms.py
@@ -102,7 +102,7 @@ def run():
     args = parser.parse_args()
     model_dir = os.path.join(args.model, config.train_ms_config.model)
     if not os.path.exists(model_dir):
-        os.makedirs(model_dir)
+        os.makedirs(model_dir, exist_ok=True)
     hps = utils.get_hparams_from_file(args.config)
     hps.model_dir = model_dir
     # 比较路径是否相同