fix save model name

2020-06-25 17:00:40 +00:00 · 2020-06-25 17:00:40 +00:00 · d09032d439
parent 232b742946
commit d09032d439
1 changed files with 1 additions and 2 deletions
--- a/tools/train.py
+++ b/tools/train.py
@ -134,8 +134,7 @@ def main(args):

                        model_path = os.path.join(config.model_save_dir,
                                                  config.ARCHITECTURE["name"])
-                        save_model(train_prog, model_path,
-                                   "best_model_in_epoch_" + str(epoch_id))
+                        save_model(train_prog, model_path, "best_model")

            # 3. save the persistable model
            if epoch_id % config.save_interval == 0: