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 |
This repository has been archived by the owner on May 19, 2022. It is now read-only.