mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

[Refactor]: Change scheduler to param_scheduler (#121)

Browse Source 
* [Refactor]: Change scheduler to param_scheduler

* [Fix]: Fix UT of param scheduler hook

Co-authored-by: Your <you@example.com>
This commit is contained in:
Yuan Liu 2022-03-12 10:47:06 +08:00 committed by GitHub
parent 61fecabea6
commit 755f8b5b59
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmengine/hooks/param_scheduler_hook.py
View File

@ -31,7 +31,7 @@ class ParamSchedulerHook(Hook):
In order to keep this interface consistent with other hooks, we
keep ``data_batch`` here. Defaults to None.
"""
for scheduler in runner.schedulers:
for scheduler in runner.param_schedulers: # type: ignore
if not scheduler.by_epoch:
scheduler.step()
@ -41,6 +41,6 @@ class ParamSchedulerHook(Hook):
Args:
runner (Runner): The runner of the training process.
"""
for scheduler in runner.schedulers:
for scheduler in runner.param_schedulers: # type: ignore
if scheduler.by_epoch:
scheduler.step()

4
tests/test_hook/test_param_scheduler_hook.py
View File

@ -12,7 +12,7 @@ class TestParamSchedulerHook:
scheduler = Mock()
scheduler.step = Mock()
scheduler.by_epoch = False
Runner.schedulers = [scheduler]
Runner.param_schedulers = [scheduler]
Hook.after_train_iter(Runner)
scheduler.step.assert_called()
@ -22,6 +22,6 @@ class TestParamSchedulerHook:
scheduler = Mock()
scheduler.step = Mock()
scheduler.by_epoch = True
Runner.schedulers = [scheduler]
Runner.param_schedulers = [scheduler]
Hook.after_train_epoch(Runner)
scheduler.step.assert_called()