SauravMaheshkar commited on
Commit
385fea7
·
verified ·
1 Parent(s): a38dab9

docs: update model card

Browse files
Files changed (1) hide show
  1. README.md +32 -5
README.md CHANGED
@@ -1,9 +1,36 @@
1
  ---
 
2
  tags:
3
- - model_hub_mixin
 
 
 
 
4
  ---
5
 
6
- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
7
- - Code: [More Information Needed]
8
- - Paper: [More Information Needed]
9
- - Docs: [More Information Needed]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ library_name: mlx
3
  tags:
4
+ - image-classification
5
+ - vision
6
+ datasets:
7
+ - imagenet
8
+ - imagenet-1k
9
  ---
10
 
11
+ ## Usage
12
+
13
+ ```python
14
+ from mlx_ssl.models import SimCLR
15
+
16
+
17
+ # Available Models:
18
+ # * mlx-community/simclrv1-imagenet1k-resnet50-1x
19
+ # * mlx-community/simclrv1-imagenet1k-resnet50-2x
20
+ # * mlx-community/simclrv1-imagenet1k-resnet50-4x
21
+
22
+ model = SimCLR.from_pretrained(
23
+ "mlx-community/simclrv1-imagenet1k-resnet50-1x"
24
+ )
25
+ ```
26
+
27
+ ## References
28
+
29
+ ```misc
30
+ @article{chen2020simple,
31
+ title={A Simple Framework for Contrastive Learning of Visual Representations},
32
+ author={Chen, Ting and Kornblith, Simon and Norouzi, Mohammad and Hinton, Geoffrey},
33
+ journal={arXiv preprint arXiv:2002.05709},
34
+ year={2020}
35
+ }
36
+ ```