Fix CIFAR normalization mean values

AidenDurrant / SimCLR-Pytorch

An implementation of "A Simple Framework for Contrastive Learning of Visual Representatoins" SimCLR

31 stars 9 forks source link

Fix CIFAR normalization mean values #3

Closed mark-arm closed 3 years ago

mark-arm commented 3 years ago

Original code accidentally used the standard deviation as the mean for CIFAR normalization. These values calculated from the torchvision.datasets.CIFAR10 training data.

mark-arm commented 3 years ago

No problem, great work on the repo though, it's nice to see reproduction experiments included (and they showed the normalization didn't matter very much!). If you want others to build on this definitely add a LICENSE file of some kind!