main_train.sh to loop over training part