From d09032d439609356607e4e2f6b5addf860808f1c Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Thu, 25 Jun 2020 17:00:40 +0000
Subject: [PATCH] fix save model name

---
 tools/train.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tools/train.py b/tools/train.py
index a5f765f0..3456e2ae 100644
--- 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:
-- 
GitLab