mirror of
https://github.com/open-mmlab/mmengine.git
synced 2025-06-03 21:54:44 +08:00
be9971781e
* [Fix]: Change after inter and epoch to after train iter and epoch * [Fix]: Add new UT to param scheduler hook * [Fix]: Change the type of runner in docstring to Runner Co-authored-by: Your <you@example.com>
28 lines
765 B
Python
28 lines
765 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from unittest.mock import Mock
|
|
|
|
from mmengine.hooks import ParamSchedulerHook
|
|
|
|
|
|
class TestParamSchedulerHook:
|
|
|
|
def test_after_iter(self):
|
|
Hook = ParamSchedulerHook()
|
|
Runner = Mock()
|
|
scheduler = Mock()
|
|
scheduler.step = Mock()
|
|
scheduler.by_epoch = False
|
|
Runner.schedulers = [scheduler]
|
|
Hook.after_train_iter(Runner)
|
|
scheduler.step.assert_called()
|
|
|
|
def test_after_epoch(self):
|
|
Hook = ParamSchedulerHook()
|
|
Runner = Mock()
|
|
scheduler = Mock()
|
|
scheduler.step = Mock()
|
|
scheduler.by_epoch = True
|
|
Runner.schedulers = [scheduler]
|
|
Hook.after_train_epoch(Runner)
|
|
scheduler.step.assert_called()
|