From 441beb50a75020e579969e9e2d4b38cdcfe7e646 Mon Sep 17 00:00:00 2001
From: wangguanzhong <jerrywgz@126.com>
Date: Fri, 13 May 2022 16:03:16 +0800
Subject: [PATCH] fix_evaldataset (#5972)

---
 ppdet/engine/trainer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppdet/engine/trainer.py b/ppdet/engine/trainer.py
index 5aaabcacb..2ae67bd91 100644
--- a/ppdet/engine/trainer.py
+++ b/ppdet/engine/trainer.py
@@ -339,7 +339,8 @@ class Trainer(object):
         assert self.mode == 'train', "Model not in 'train' mode"
         Init_mark = False
         if validate:
-            self.cfg.EvalDataset = create("EvalDataset")()
+            self.cfg['EvalDataset'] = self.cfg.EvalDataset = create(
+                "EvalDataset")()
 
         model = self.model
         if self.cfg.get('fleet', False):
-- 
GitLab