fix save model name

d09032d4 · littletomatodonkey · 232b7429 · d09032d4
显示空白变更内容
内联并排

Showing with 1 addition and 2 deletion

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

未找到文件。
--- 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,
+                        save_model(train_prog, model_path, "best_model")
-                                   "best_model_in_epoch_" + str(epoch_id))
            # 3. save the persistable model
            if epoch_id % config.save_interval == 0: