train.sh