diff --git a/tools/train.py b/tools/train.py
index bb2ee9fa2e480587c99d37498a22bd13e38255b4..32af634966bb7e76991a0b9e29dbcb36168b37fe 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -105,7 +105,7 @@ def main(args):
                                                   config.ARCHITECTURE["name"])
                         save_model(net, optimizer, model_path, "best_model")
                 message = "The best top1 acc {:.5f}, in epoch: {:d}".format(
-                    best_top1_acc, epoch_id)
+                    best_top1_acc, best_top1_epoch)
                 logger.info("{:s}".format(logger.coloring(message, "RED")))
 
             # 3. save the persistable model