From fee238ff8cb0576e7dfdbc0f76916c50bdf882fd Mon Sep 17 00:00:00 2001
From: Tingquan Gao <35441050@qq.com>
Date: Tue, 14 Mar 2023 16:16:40 +0800
Subject: [PATCH] Revert "debug"

This reverts commit 106ff27010a86e5fb6ac5a8fd31ad8335dcc90d2.
---
 ppcls/engine/train/classification.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppcls/engine/train/classification.py b/ppcls/engine/train/classification.py
index 6f84133b3..9074a6b49 100644
--- a/ppcls/engine/train/classification.py
+++ b/ppcls/engine/train/classification.py
@@ -258,7 +258,7 @@ class ClassTrainer(object):
         return None
 
     def _build_ema_model(self):
-        if "EMA" in self.config:
+        if "EMA" in self.config and self.mode == "train":
             model_ema = ExponentialMovingAverage(
                 self.model, self.config['EMA'].get("decay", 0.9999))
             self.best_metric["metric_ema"] = 0