fix coslr (#32)

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-08-13 10:45:41 +08:00 · 2020-08-13 10:45:41 +08:00 · 414ede4ca4
parent 608cea2469
commit 414ede4ca4
2 changed files with 2 additions and 2 deletions
--- a/configs/_base_/schedules/imagenet_bs2048_coslr.py
+++ b/configs/_base_/schedules/imagenet_bs2048_coslr.py
@ -4,7 +4,7 @@ optimizer = dict(
 optimizer_config = dict(grad_clip=None)
 # learning policy
 lr_config = dict(
-    policy='CosineAnealing',
+    policy='CosineAnnealing',
    min_lr=0,
    warmup='linear',
    warmup_iters=2500,
--- a/configs/_base_/schedules/imagenet_bs256_coslr.py
+++ b/configs/_base_/schedules/imagenet_bs256_coslr.py
@ -2,5 +2,5 @@
 optimizer = dict(type='SGD', lr=0.1, momentum=0.9, weight_decay=0.0001)
 optimizer_config = dict(grad_clip=None)
 # learning policy
-lr_config = dict(policy='CosineAnealing', min_lr=0)
+lr_config = dict(policy='CosineAnnealing', min_lr=0)
 total_epochs = 100