fix evaldataset (#5978)

67742521 · wangguanzhong · GitHub · 1480c8ee · 67742521
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

ppdet/engine/trainer.py ppdet/engine/trainer.py +2 -1

未找到文件。
--- a/ppdet/engine/trainer.py
+++ b/ppdet/engine/trainer.py
@@ -376,7 +376,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")()

        sync_bn = (getattr(self.cfg, 'norm_type', None) == 'sync_bn' and
                   self.cfg.use_gpu and self._nranks > 1)