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 |
deepankarvarma/SimCLRV2-Tensorflow-Checkpoints-Conversion-To-Pytorch
Code for conversion of google research SimCLR V2 Tensorflow checkpoints to Pytorch ones. SimCLRV2 Repo:- https://github.com/google-research/simclr
PythonGPL-3.0