timm
/

Image Classification
timm
PyTorch
Safetensors
rwightman HF staff commited on
Commit
84a41ee
1 Parent(s): 89afddd

Update model config and README

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -37,8 +37,8 @@ Recipe details:
37
  - **Model Type:** Image classification / feature backbone
38
  - **Model Stats:**
39
  - Params (M): 60.4
40
- - GMACs: 15.5
41
- - Activations (M): 18.1
42
  - Image size: 256 x 256
43
  - **Papers:**
44
  - Vision Transformers Need Registers: https://arxiv.org/abs/2309.16588
@@ -149,6 +149,7 @@ output = model.forward_head(output, pre_logits=True)
149
  | [vit_medium_patch16_reg4_gap_256.sbb_in1k](https://huggingface.co/timm/vit_medium_patch16_reg4_gap_256.sbb_in1k) | 83.47 | 96.622 | 38.88 | 256 |
150
  | [vit_medium_patch16_reg1_gap_256.sbb_in1k](https://huggingface.co/timm/vit_medium_patch16_reg1_gap_256.sbb_in1k) | 83.462 | 96.548 | 38.88 | 256 |
151
  | [vit_little_patch16_reg4_gap_256.sbb_in1k](https://huggingface.co/timm/vit_little_patch16_reg4_gap_256.sbb_in1k) | 82.514 | 96.262 | 22.52 | 256 |
 
152
  | [vit_pwee_patch16_reg1_gap_256.sbb_in1k](https://huggingface.co/timm/vit_pwee_patch16_reg1_gap_256.sbb_in1k) | 80.072 | 95.136 | 15.25 | 256 |
153
 
154
  ## Citation
 
37
  - **Model Type:** Image classification / feature backbone
38
  - **Model Stats:**
39
  - Params (M): 60.4
40
+ - GMACs: 16.5
41
+ - Activations (M): 28.2
42
  - Image size: 256 x 256
43
  - **Papers:**
44
  - Vision Transformers Need Registers: https://arxiv.org/abs/2309.16588
 
149
  | [vit_medium_patch16_reg4_gap_256.sbb_in1k](https://huggingface.co/timm/vit_medium_patch16_reg4_gap_256.sbb_in1k) | 83.47 | 96.622 | 38.88 | 256 |
150
  | [vit_medium_patch16_reg1_gap_256.sbb_in1k](https://huggingface.co/timm/vit_medium_patch16_reg1_gap_256.sbb_in1k) | 83.462 | 96.548 | 38.88 | 256 |
151
  | [vit_little_patch16_reg4_gap_256.sbb_in1k](https://huggingface.co/timm/vit_little_patch16_reg4_gap_256.sbb_in1k) | 82.514 | 96.262 | 22.52 | 256 |
152
+ | [vit_wee_patch16_reg1_gap_256.sbb_in1k](https://huggingface.co/timm/vit_wee_patch16_reg1_gap_256.sbb_in1k) | 80.258 | 95.360 | 13.42 | 256 |
153
  | [vit_pwee_patch16_reg1_gap_256.sbb_in1k](https://huggingface.co/timm/vit_pwee_patch16_reg1_gap_256.sbb_in1k) | 80.072 | 95.136 | 15.25 | 256 |
154
 
155
  ## Citation