[Fix]: Fix UT
parent
0a2a811acf
commit
8d18f42c32
|
@ -1,38 +0,0 @@
|
||||||
# Copyright (c) OpenMMLab. All rights reserved.
|
|
||||||
from unittest.mock import MagicMock
|
|
||||||
|
|
||||||
from mmselfsup.core.hooks import StepFixCosineAnnealingLrUpdaterHook
|
|
||||||
|
|
||||||
|
|
||||||
def test_cosine_annealing_hook():
|
|
||||||
lr_config = dict(
|
|
||||||
min_lr=1e-5,
|
|
||||||
warmup='linear',
|
|
||||||
warmup_iters=10,
|
|
||||||
warmup_ratio=1e-4,
|
|
||||||
warmup_by_epoch=True,
|
|
||||||
by_epoch=False)
|
|
||||||
lr_annealing_hook = StepFixCosineAnnealingLrUpdaterHook(**lr_config)
|
|
||||||
lr_annealing_hook.regular_lr = [1.0]
|
|
||||||
lr_annealing_hook.warmup_iters = 10
|
|
||||||
|
|
||||||
# test get_warmup_lr
|
|
||||||
lr = lr_annealing_hook.get_warmup_lr(1)
|
|
||||||
assert isinstance(lr, list)
|
|
||||||
|
|
||||||
# test get_lr
|
|
||||||
# by_epoch = False
|
|
||||||
runner = MagicMock()
|
|
||||||
runner.iter = 10
|
|
||||||
runner.max_iters = 1000
|
|
||||||
lr = lr_annealing_hook.get_lr(runner, 1.5)
|
|
||||||
assert isinstance(lr, float)
|
|
||||||
|
|
||||||
# by_epoch = True
|
|
||||||
lr_annealing_hook.by_epoch = True
|
|
||||||
runner.epoch = 10
|
|
||||||
runner.max_epochs = 10
|
|
||||||
runner.data_loader = MagicMock()
|
|
||||||
runner.data_loader.__len__ = MagicMock(return_value=10)
|
|
||||||
lr = lr_annealing_hook.get_lr(runner, 1.5)
|
|
||||||
assert isinstance(lr, float)
|
|
Loading…
Reference in New Issue