From 4ebd7e21ea119c60104215f9d9d837e318c6c9aa Mon Sep 17 00:00:00 2001 From: litagin02 Date: Mon, 1 Jan 2024 00:08:56 +0900 Subject: [PATCH] Fix colab bug --- train_ms.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train_ms.py b/train_ms.py index 544a765..46c2d11 100644 --- 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)