minor weight decay specification bug fix