ai-photo-gallery / configs /resnet /resnet50_8xb16_cifar100.py
KyanChen's picture
init
f549064
raw
history blame
432 Bytes
_base_ = [
'../_base_/models/resnet50_cifar.py',
'../_base_/datasets/cifar100_bs16.py',
'../_base_/schedules/cifar10_bs128.py',
'../_base_/default_runtime.py',
]
# model settings
model = dict(head=dict(num_classes=100))
# schedule settings
optim_wrapper = dict(optimizer=dict(weight_decay=0.0005))
param_scheduler = dict(
type='MultiStepLR',
by_epoch=True,
milestones=[60, 120, 160],
gamma=0.2,
)