nielsr HF staff commited on
Commit
a1a6cb6
1 Parent(s): 7446dfb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -5
README.md CHANGED
@@ -25,7 +25,7 @@ Disclaimer: The team releasing RegNet did not write a model card for this model
25
 
26
  ## Model description
27
 
28
- The authors trained [RegNets](https://huggingface.co/?models=regnet) models in a self-supervised fashion on bilion of random images from the internet. This model is later finetuned on ImageNet
29
 
30
  ![model image](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/regnet_architecture.png)
31
 
@@ -46,8 +46,8 @@ Here is how to use this model:
46
  >>> dataset = load_dataset("huggingface/cats-image")
47
  >>> image = dataset["test"]["image"][0]
48
 
49
- >>> feature_extractor = AutoFeatureExtractor.from_pretrained("zuppif/regnet-y-040")
50
- >>> model = RegNetForImageClassification.from_pretrained("zuppif/regnet-y-040")
51
 
52
  >>> inputs = feature_extractor(image, return_tensors="pt")
53
 
@@ -60,6 +60,4 @@ Here is how to use this model:
60
  'tabby, tabby cat'
61
  ```
62
 
63
-
64
-
65
  For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/regnet).
 
25
 
26
  ## Model description
27
 
28
+ The authors trained [RegNets](https://huggingface.co/?models=regnet) models in a self-supervised fashion on a billion uncurated Instagram images. This model is later fine-tuned on ImageNet.
29
 
30
  ![model image](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/regnet_architecture.png)
31
 
 
46
  >>> dataset = load_dataset("huggingface/cats-image")
47
  >>> image = dataset["test"]["image"][0]
48
 
49
+ >>> feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/regnet-y-320-seer-in1k")
50
+ >>> model = RegNetForImageClassification.from_pretrained("facebook/regnet-y-320-seer-in1k")
51
 
52
  >>> inputs = feature_extractor(image, return_tensors="pt")
53
 
 
60
  'tabby, tabby cat'
61
  ```
62
 
 
 
63
  For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/regnet).