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