Delete train_model.h