Fixed the comment in the Weight decay class.