Fix colab bug

2024-01-01 00:08:56 +09:00
parent d0df0f3526
commit 4ebd7e21ea
1 changed files with 1 additions and 0 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -114,6 +114,7 @@ def run():
    ):
        with open(args.config, "r", encoding="utf-8") as f:
            data = f.read()
+        os.makedirs(os.path.dirname(config.train_ms_config.config_path), exist_ok=True)
        with open(config.train_ms_config.config_path, "w", encoding="utf-8") as f:
            f.write(data)