More details about train_model.py.