simCLR pretrained model load error

[kfarivar]

Hi
there seems to be a problem loading simCLR for CIFAR10:

from pl_bolts.models.self_supervised import SimCLR

weight_path = 'https://pl-bolts-weights.s3.us-east-2.amazonaws.com/simclr/simclr-cifar10-v1-exp12_87_52/epoch%3D960.ckpt'
simclr = SimCLR.load_from_checkpoint(weight_path, strict=False)

simclr.freeze()

output:

RuntimeError: Error(s) in loading state_dict for SimCLR:
size mismatch for encoder.conv1.weight: copying a param with shape torch.Size([64, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([64, 3, 7, 7]).
size mismatch for projection.model.3.weight: copying a param with shape torch.Size([2048]) from checkpoint, the shape in current model is torch.Size([128, 2048]).

Also, the checkpoints in the docs are wrong (they are the same as the ones for imagenet).

and Shouldn't we set some flags when initializing simCLR for CIFAR10 since the arch is different than imagenet ? (e.g. the kernel size of the initial convolutional layer for CIFAR10 is smaller.)