add vloss to train.sh