Added dynamic image sizes to CIFAR