Compute the mean and std about the direct datasets

[SWHL]

It may be something wrong on the line 42 in the utils.py

SE-Net-CIFAR/utils.py

Line 42 in d34fb41

mean.div_(len(dataset))

I think it should be the following code:

    mean.div_(len(dataloader))
    std.div_(len(dataloader))