21 lines
513 B
Python
21 lines
513 B
Python
_base_ = [
|
|
'../_base_/models/mobileone/mobileone_s0.py',
|
|
'../_base_/datasets/imagenet_bs32_pil_resize.py',
|
|
'../_base_/schedules/imagenet_bs256_coslr_coswd_300e.py',
|
|
'../_base_/default_runtime.py'
|
|
]
|
|
|
|
# schedule settings
|
|
optim_wrapper = dict(paramwise_cfg=dict(norm_decay_mult=0.))
|
|
|
|
val_dataloader = dict(batch_size=256)
|
|
test_dataloader = dict(batch_size=256)
|
|
|
|
custom_hooks = [
|
|
dict(
|
|
type='EMAHook',
|
|
momentum=5e-4,
|
|
priority='ABOVE_NORMAL',
|
|
update_buffers=True)
|
|
]
|