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

Download this file

11 lines (10 with data), 281 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=[20, 40])
total_epochs = 50