reinit_weights function

Generic function to re-intialize model weights