train model on model weights to produce better initializations