allow setting custom weight decay