[6d389a]: / configs / _base_ / schedules / sgd_100e.py

Download this file

11 lines (10 with data), 282 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
# optimizer
optimizer = dict(
type='SGD',
lr=0.01, # this lr is used for 8 gpus
momentum=0.9,
weight_decay=0.0001)
optimizer_config = dict(grad_clip=dict(max_norm=40, norm_type=2))
# learning policy
lr_config = dict(policy='step', step=[40, 80])
total_epochs = 100