script to run training on cifar added