mmpretrain/configs/_base_/schedules/imagenet_bs256_epochstep.py

10 lines
328 B
Python
Raw Normal View History

# optimizer
optimizer = dict(type='SGD', lr=0.045, momentum=0.9, weight_decay=0.00004)
# learning policy
2022-05-23 09:31:57 +00:00
param_scheduler = dict(type='StepLR', by_epoch=True, step_size=1, gamma=0.98)
# train, val, test setting
train_cfg = dict(by_epoch=True, max_epochs=300)
val_cfg = dict(interval=1) # validate every epoch
test_cfg = dict()