modified cifar10_train.sh