File size: 698 Bytes
20df252 6198e4b 20df252 6198e4b 20df252 6198e4b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
---
library_name: mlx
tags:
- image-classification
- vision
datasets:
- imagenet
- imagenet-1k
---
## Usage
```python
from mlx_ssl.models import SimCLR
# Available Models:
# * mlx-community/simclrv1-imagenet1k-resnet50-1x
# * mlx-community/simclrv1-imagenet1k-resnet50-2x
# * mlx-community/simclrv1-imagenet1k-resnet50-4x
model = SimCLR.from_pretrained(
"mlx-community/simclrv1-imagenet1k-resnet50-1x"
)
```
## References
```misc
@article{chen2020simple,
title={A Simple Framework for Contrastive Learning of Visual Representations},
author={Chen, Ting and Kornblith, Simon and Norouzi, Mohammad and Hinton, Geoffrey},
journal={arXiv preprint arXiv:2002.05709},
year={2020}
}
``` |