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,
-                                   "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: