Update subsample_train.sh