1. Add the train.sh, contains the command used to train