Merge pull request #862 from lzjpaul/21-7-6-train-script

train script for cifar 10 distributed training