Set weight decay back to 0.1