Revert "debug"

This reverts commit 106ff270.

Revert "debug"
This reverts commit 106ff270.
fee238ff · Tingquan Gao · 50284e7b · fee238ff
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppcls/engine/train/classification.py ppcls/engine/train/classification.py +1 -1

未找到文件。
--- 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