Better Training using Weight-Constrained Stochastic Dynamics