timm
/

Image Classification
timm
PyTorch
Safetensors
rwightman HF staff commited on
Commit
c3757b4
1 Parent(s): c51b452

Update model config and README

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -36,8 +36,8 @@ Recipe details:
36
  - **Model Type:** Image classification / feature backbone
37
  - **Model Stats:**
38
  - Params (M): 15.3
39
- - GMACs: 3.8
40
- - Activations (M): 10.6
41
  - Image size: 256 x 256
42
  - **Papers:**
43
  - Vision Transformers Need Registers: https://arxiv.org/abs/2309.16588
@@ -147,6 +147,7 @@ output = model.forward_head(output, pre_logits=True)
147
  | [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 |
148
  | [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 |
149
  | [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 |
 
150
  | [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 |
151
 
152
  ## Citation
 
36
  - **Model Type:** Image classification / feature backbone
37
  - **Model Stats:**
38
  - Params (M): 15.3
39
+ - GMACs: 4.4
40
+ - Activations (M): 15.9
41
  - Image size: 256 x 256
42
  - **Papers:**
43
  - Vision Transformers Need Registers: https://arxiv.org/abs/2309.16588
 
147
  | [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 |
148
  | [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 |
149
  | [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 |
150
+ | [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 |
151
  | [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 |
152
 
153
  ## Citation