fixes to weight decay
revert to weight shrinking after a nan in minibatch