mlx-community/simclrv1-imagenet1k-resnet50-2x

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