Commit 16c3f6e1 authored by zhangwenwei's avatar zhangwenwei
Browse files

Merge branch 'lr_config' into 'master'

fix lr_config problem

See merge request open-mmlab/mmdet.3d!51
parents 37c87f22 be08155b
...@@ -255,13 +255,11 @@ optimizer = dict(type='AdamW', lr=0.003, betas=(0.95, 0.99), weight_decay=0.01) ...@@ -255,13 +255,11 @@ optimizer = dict(type='AdamW', lr=0.003, betas=(0.95, 0.99), weight_decay=0.01)
# max_norm=10 is better for SECOND # max_norm=10 is better for SECOND
optimizer_config = dict(grad_clip=dict(max_norm=35, norm_type=2)) optimizer_config = dict(grad_clip=dict(max_norm=35, norm_type=2))
lr_config = dict( lr_config = dict(
policy='cosine', policy='CosineAnealing',
warmup='linear', warmup='linear',
warmup_iters=1000, warmup_iters=1000,
warmup_ratio=1.0 / 10, warmup_ratio=1.0 / 10,
target_lr=1e-5, min_lr_ratio=1e-5)
as_ratio=True,
)
momentum_config = None momentum_config = None
checkpoint_config = dict(interval=1) checkpoint_config = dict(interval=1)
# yapf:disable # yapf:disable
......
...@@ -207,13 +207,11 @@ optimizer = dict( ...@@ -207,13 +207,11 @@ optimizer = dict(
weight_decay=0.001) weight_decay=0.001)
optimizer_config = dict(grad_clip=dict(max_norm=10, norm_type=2)) optimizer_config = dict(grad_clip=dict(max_norm=10, norm_type=2))
lr_config = dict( lr_config = dict(
policy='cosine', policy='CosineAnealing',
warmup='linear', warmup='linear',
warmup_iters=1000, warmup_iters=1000,
warmup_ratio=1.0 / 10, warmup_ratio=1.0 / 10,
target_lr=1e-5, min_lr_ratio=1e-5)
as_ratio=True,
)
momentum_config = None momentum_config = None
checkpoint_config = dict(interval=1) checkpoint_config = dict(interval=1)
# yapf:disable # yapf:disable
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment