mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
7977dc8e2d
* Update tnt config * Rename config files * Update docs * Update metafile in dev_scripts * Fix some files * Remove fp16 metafile and README. * Fix names in metafiles.
11 lines
352 B
Python
11 lines
352 B
Python
_base_ = [
|
|
'../_base_/models/resnet50_cifar.py',
|
|
'../_base_/datasets/cifar100_bs16.py',
|
|
'../_base_/schedules/cifar10_bs128.py', '../_base_/default_runtime.py'
|
|
]
|
|
|
|
model = dict(head=dict(num_classes=100))
|
|
|
|
optimizer = dict(type='SGD', lr=0.1, momentum=0.9, weight_decay=0.0005)
|
|
lr_config = dict(policy='step', step=[60, 120, 160], gamma=0.2)
|