weight decay now accepts a single coefficient