diff --git a/tests/test_core/test_delivers/test_function_outputs_deliver.py b/tests/test_core/test_delivers/test_function_outputs_deliver.py index 9f59f695c..ad9a7f5f8 100644 --- a/tests/test_core/test_delivers/test_function_outputs_deliver.py +++ b/tests/test_core/test_delivers/test_function_outputs_deliver.py @@ -142,6 +142,7 @@ def test_ema_hook(self): num_workers=0), val_evaluator=evaluator, work_dir=self.temp_dir.name, + default_scope='mmrazor', optim_wrapper=OptimWrapper( torch.optim.Adam(ToyModel().parameters())), train_cfg=dict(by_epoch=True, max_epochs=2, val_interval=1), diff --git a/tests/test_core/test_recorders/test_func_inputs_recorder.py b/tests/test_core/test_recorders/test_func_inputs_recorder.py index 4b8339181..9a3dbc4d4 100644 --- a/tests/test_core/test_recorders/test_func_inputs_recorder.py +++ b/tests/test_core/test_recorders/test_func_inputs_recorder.py @@ -117,6 +117,7 @@ def test_ema_hook(self): num_workers=0), val_evaluator=evaluator, work_dir=self.temp_dir.name, + default_scope='mmrazor', optim_wrapper=OptimWrapper( torch.optim.Adam(ToyModel().parameters())), train_cfg=dict(by_epoch=True, max_epochs=2, val_interval=1),