add weight decay module