diff --git a/ppcls/engine/trainer.py b/ppcls/engine/trainer.py index 53235cc0ee7bd670dec13ecbd40fb726423d1f00..08fe90bf906fef624e782dcd54612512acc7dd00 100644 --- a/ppcls/engine/trainer.py +++ b/ppcls/engine/trainer.py @@ -236,7 +236,7 @@ class Trainer(object): model_name=self.config["Arch"]["name"], prefix="best_model") logger.info("[Eval][Epoch {}][best metric: {}]".format( - epoch_id, acc)) + epoch_id, best_metric["metric"])) self.model.train() # save model