SauravMaheshkar's picture
docs: update model card
6198e4b verified
metadata
library_name: mlx
tags:
  - image-classification
  - vision
datasets:
  - imagenet
  - imagenet-1k

Usage

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

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