To train the model, please run run.sh.