|
|
--- |
|
|
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} |
|
|
} |
|
|
``` |
