fix priority for eval hook

open-mmlab · Sep 29, 2021 · ce6e4eb · ce6e4eb
1 parent 432850b
commit ce6e4eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmgen/apis/train.py b/mmgen/apis/train.py
@@ -172,9 +172,9 @@ def train_model(model,
         val_dataloader = build_dataloader(
             val_dataset, dist=distributed, **val_loader_cfg)
         eval_cfg = deepcopy(cfg.get('evaluation'))
+        priority = eval_cfg.pop('priority', 'LOW')
         eval_cfg.update(dict(dist=distributed, dataloader=val_dataloader))
         eval_hook = build_from_cfg(eval_cfg, HOOKS)
-        priority = eval_cfg.pop('priority', 'NORMAL')
         runner.register_hook(eval_hook, priority=priority)
 
     # user-defined hooks