Yan
@gsp-27, when I read the [source code](https://github.com/pytorch/vision/blob/master/torchvision/datasets/cifar.py#L121) of loading CIFAR10 dataset, I found that the param of `transform` is just used in the function `__getitem__(self, index)`, which means that only...
@zhaofenqiang Following my suggestion, you should modify [compute-cifar10-mean.py#L31](https://github.com/bamos/densenet.pytorch/blob/master/compute-cifar10-mean.py#L31) to `pixels = data[:,:,:,i].ravel()`, and the result is same as [demo.py](https://github.com/gpleiss/efficient_densenet_pytorch/blob/8e788a24fa135b6a6542dc312dfcad1f8621a832/demo.py).