From ce6e4eb748284cb2f41da767d20dcb9d07613957 Mon Sep 17 00:00:00 2001 From: yangyifei Date: Wed, 29 Sep 2021 17:24:36 +0800 Subject: [PATCH] fix priority for eval hook --- mmgen/apis/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmgen/apis/train.py b/mmgen/apis/train.py index a9376dd76..a155eca31 100644 --- 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