mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
4969830c8a
* add switch hook and UTs * update doc * update doc * fix lint * fix ci * fix ci * fix typo * fix ci * update configs names * update configs * update configs * update links * update readme * update vis_scheduler * update metafile * update configs * rebase * fix ci * rebase
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)
|
|
]
|