adam_20e.py 225 Bytes
Newer Older
unknown's avatar
unknown committed
1
2
3
4
5
6
7
# optimizer
optimizer = dict(
    type='Adam', lr=0.01, weight_decay=0.00001)  # this lr is used for 1 gpus
optimizer_config = dict(grad_clip=None)
# learning policy
lr_config = dict(policy='step', step=10)
total_epochs = 20