fix save model

2020-04-16 08:26:10 +00:00 · 2020-04-16 08:26:10 +00:00 · 0b39b19006
parent 3b69b6c854
commit 0b39b19006
1 changed files with 1 additions and 1 deletions
--- 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)