Merge pull request #894 from NLGithubWP/update-distributed-train-cnn-dev

Update data augmentation implementation for cifar_distributed_cnn exa…