EasyCV/configs/detection/dino/dino_schedule_1x.py
tuofeilun b198c5a81f
support DINO algo (#144)
dino_4sc_r50_12e:48.71
dino_4sc_r50_24e:50.53
dino_4sc_r50_36e:50.69
dino_4sc_swinl_12e: 56.86
dino_4sc_swinl_36e: 58.04
dino_5sc_swinl_36e: 58.47
2022-08-31 15:18:11 +08:00

20 lines
429 B
Python

_base_ = 'configs/base.py'
checkpoint_config = dict(interval=10)
# optimizer
paramwise_options = {
'backbone': dict(lr_mult=0.1),
}
optimizer = dict(
type='AdamW',
lr=1e-4,
weight_decay=1e-4,
paramwise_options=paramwise_options)
optimizer_config = dict(grad_clip=dict(max_norm=0.1, norm_type=2))
# learning policy
lr_config = dict(policy='step', step=[11])
total_epochs = 12
find_unused_parameters = False