forgot to set some weight decays to zero