lr: 0.01
weight_decay: 0.0005
