Pretrained SimCLRv2 models in Pytorch
python download.py r152_3x_sk1
python convert.py r152_3x_sk1/model.ckpt-250228 [--ema]
python verify.py r152_3x_sk1.pth
Model | Pytorch | TF |
---|---|---|
r50_1x_sk0 | 70.97 | 71.7 |
r50_1x_sk1 | 73.79 | 74.6 |
r152_3x_sk1 | 79.12 | 79.8 |