mmpretrain/configs/mobilenet_v3/mobilenet-v3-small_8xb16_cifar10.py
2022-07-18 10:53:28 +08:00

10 lines
334 B
Python

_base_ = [
'../_base_/models/mobilenet-v3-small_8xb16_cifar.py',
'../_base_/datasets/cifar10_bs16.py',
'../_base_/schedules/cifar10_bs128.py', '../_base_/default_runtime.py'
]
param_scheduler = dict(
type='MultiStepLR', by_epoch=True, milestones=[120, 170], gamma=0.1)
train_cfg = dict(by_epoch=True, max_epochs=200)