alihjt commited on
Commit
f9d6ffc
·
verified ·
1 Parent(s): 48cdb88

Sanitize EMA checkpoint and align paper metrics

Browse files
Files changed (2) hide show
  1. README.md +9 -8
  2. progresvit_192_240.pth.tar +2 -2
README.md CHANGED
@@ -1,7 +1,5 @@
1
  ---
2
  license: mit
3
- library_name: timm
4
- pipeline_tag: image-classification
5
  tags:
6
  - image-classification
7
  - vision-transformer
@@ -11,13 +9,12 @@ datasets:
11
  - ILSVRC/imagenet-1k
12
  ---
13
 
14
- # progresvit-deit-s-192-240-imagenet1k
15
 
16
- EMA checkpoint for the DeiT-S ProgResViT model trained on ImageNet-1K.
17
 
18
- - Resolution schedule: `192 -> 240`
19
- - Training: Standard
20
- - Full-path top-1 accuracy: 82.202%
21
  - Full-path compute: 6.267 GMACs
22
 
23
  ## Download
@@ -31,4 +28,8 @@ checkpoint = hf_hub_download(
31
  )
32
  ```
33
 
34
- Training and evaluation code: https://github.com/ds-kiel/ProgResViT
 
 
 
 
 
1
  ---
2
  license: mit
 
 
3
  tags:
4
  - image-classification
5
  - vision-transformer
 
9
  - ILSVRC/imagenet-1k
10
  ---
11
 
12
+ # ProgResViT DeiT-S (192240)
13
 
14
+ EMA inference checkpoint for the DeiT-S ProgResViT model trained on ImageNet-1K.
15
 
16
+ - Training: standard
17
+ - Paper-reported full-path top-1 accuracy: 82.206%
 
18
  - Full-path compute: 6.267 GMACs
19
 
20
  ## Download
 
28
  )
29
  ```
30
 
31
+ The file contains EMA weights and public architecture metadata only. Use it with the [ProgResViT repository](https://github.com/ds-kiel/ProgResViT):
32
+
33
+ ```bash
34
+ DATA_DIR=/path/to/imagenet CONFIG=192_240 bash scripts/validate.sh
35
+ ```
progresvit_192_240.pth.tar CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8d40db265b8da68a9cf16cd3c20ae29cd40ab03d02c02e939b9619a8ec6ca130
3
- size 396242239
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fab254d778b2c47b2fb6909169205564f7696336f86e3db21309f880fb25617f
3
+ size 99032753