change ModelTrainer