fix save model

0b39b190 · littletomatodonkey · 3b69b6c8 · 0b39b190
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tools/train.py tools/train.py +1 -1

未找到文件。
--- a/tools/train.py
+++ b/tools/train.py
@@ -96,7 +96,7 @@ def main(args):

        if epoch_id % config.save_interval == 0:
            model_path = os.path.join(config.model_save_dir,
-                                      config.ARCHITECTURE)
+                                      config.ARCHITECTURE["name"])
            save_model(train_prog, model_path, epoch_id)