From d49e1296d95270afed17ed0aba9a056b479c9656 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Sat, 12 Jun 2021 03:54:56 +0800
Subject: [PATCH] Update trainer.py

---
 ppcls/engine/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppcls/engine/trainer.py b/ppcls/engine/trainer.py
index 53235cc0..08fe90bf 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
-- 
GitLab