Using hyperparameter-search in Trainer

Use model_init as a keyword rather than model in Trainer.

Don’t do this: Trainer(model=model_init)
Do this: Trainer(model_init=model_init)