Update visionTrain.sh