EduardoPacheco commited on
Commit
3384168
1 Parent(s): 05a5d2c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -1,11 +1,11 @@
1
  ---
2
- license: cc-by-nc-4.0
3
- library_name: transformers
4
  datasets:
5
  - imagenet-1k
6
- pipeline_tag: image-classification
7
  language:
8
  - en
 
 
 
9
  ---
10
  # Hiera Model (Tiny, fine-tuned on IN1K)
11
 
@@ -27,7 +27,7 @@ We show that a lot of this bulk is actually _unnecessary_. Instead of manually a
27
 
28
  ## Intended uses & limitations
29
 
30
- Hiera can be used for image classification, feature extraction or masked image modeling. This checkpoint in specific is intended for **Image Classificaiton**
31
 
32
  ### How to use
33
 
@@ -55,6 +55,7 @@ predicted_id = outputs.logits.argmax(dim=-1).item()
55
  predicted_class = model.config.id2label[predicted_id] # 'tabby, tabby cat'
56
  ```
57
 
 
58
  ### BibTeX entry and citation info
59
  If you use Hiera or this code in your work, please cite:
60
  ```
@@ -63,5 +64,4 @@ If you use Hiera or this code in your work, please cite:
63
  author={Ryali, Chaitanya and Hu, Yuan-Ting and Bolya, Daniel and Wei, Chen and Fan, Haoqi and Huang, Po-Yao and Aggarwal, Vaibhav and Chowdhury, Arkabandhu and Poursaeed, Omid and Hoffman, Judy and Malik, Jitendra and Li, Yanghao and Feichtenhofer, Christoph},
64
  journal={ICML},
65
  year={2023}
66
- }
67
- ```
 
1
  ---
 
 
2
  datasets:
3
  - imagenet-1k
 
4
  language:
5
  - en
6
+ library_name: transformers
7
+ license: cc-by-nc-4.0
8
+ pipeline_tag: image-classification
9
  ---
10
  # Hiera Model (Tiny, fine-tuned on IN1K)
11
 
 
27
 
28
  ## Intended uses & limitations
29
 
30
+ Hiera can be used for image classification, feature extraction or masked image modeling. This checkpoint in specific is intended for **Image Classificaiton**.
31
 
32
  ### How to use
33
 
 
55
  predicted_class = model.config.id2label[predicted_id] # 'tabby, tabby cat'
56
  ```
57
 
58
+
59
  ### BibTeX entry and citation info
60
  If you use Hiera or this code in your work, please cite:
61
  ```
 
64
  author={Ryali, Chaitanya and Hu, Yuan-Ting and Bolya, Daniel and Wei, Chen and Fan, Haoqi and Huang, Po-Yao and Aggarwal, Vaibhav and Chowdhury, Arkabandhu and Poursaeed, Omid and Hoffman, Judy and Malik, Jitendra and Li, Yanghao and Feichtenhofer, Christoph},
65
  journal={ICML},
66
  year={2023}
67
+ }