update weights and logs for ViT and ViTAE

codwest · Apr 19, 2022 · 66e9fcc · 66e9fcc
1 parent 0da05a9
commit 66e9fcc
Show file tree

Hide file tree

Showing 4 changed files with 11,031 additions and 7,354 deletions.
diff --git a/README.md b/README.md
@@ -16,8 +16,9 @@ The models are trained on 4 A100 machines with 2 images per gpu, which makes a b
 | Model | Pretrain | Machine | FrameWork | Box mAP | Mask mAP | config | log | weight |
 | :----: | :----: | :----: | :----: | :----: | :----: | :----: | :----: | :----: | 
 | ViT-Base | IN1K+MAE | TPU | Mask RCNN | 51.1 | 45.5 | [config](./configs/ViTDet/ViTDet-ViT-Base-100e.py) | [log](logs/ViT-Base-TPU.log.json) | [OneDrive](https://1drv.ms/u/s!AimBgYV7JjTlgQuegyG-Z3FH2LDP?e=9ij98g) |
-| ViT-Base | IN1K+MAE | GPU | Mask RCNN | 50.9 | 45.2 | [config](./configs/ViTDet/ViTDet-ViT-Base-100e.py) | [log](logs/ViT-Base-GPU.log.json) | [OneDrive](https://1drv.ms/u/s!AimBgYV7JjTlgQyB0R1bn1h7HzSz?e=IczYEw) |
-| ViTAE-Base | IN1K+MAE | GPU | Mask RCNN | 51.2 | 45.6 | [config](configs/ViTDet/ViTDet-ViTAE-Base-100e.py) | [log](logs/ViTAE-Base-GPU.log.json) | [OneDrive](https://1drv.ms/u/s!AimBgYV7JjTlgQ0mNFX8iGvMAGmI) |
+| ViT-Base | IN1K+MAE | GPU | Mask RCNN | 51.1 | 45.4 | [config](./configs/ViTDet/ViTDet-ViT-Base-100e.py) | [log](logs/ViT-Base-GPU.log.json) | [OneDrive](https://1drv.ms/u/s!AimBgYV7JjTlgRA7Y9s2rA5NC4wn?e=QfpKJf) |
+| ViTAE-Base | IN1K+MAE | GPU | Mask RCNN | 51.6 | 45.8 | [config](configs/ViTDet/ViTDet-ViTAE-Base-100e.py) | [log](logs/ViTAE-Base-GPU.log.json) | [OneDrive](https://1drv.ms/u/s!AimBgYV7JjTlgQ--Ez4mzEnO-G5Y?e=ACfLxC) |
+| ViTAE-Small | IN1K+Sup | GPU | Mask RCNN | 45.6 | 40.1 | | [log](logs/ViTAE-S-GPU.log.json) | [weight](https://1drv.ms/u/s!AimBgYV7JjTlgQ7PorGY53K6gIGd?e=lw81U5) |
 
 ## Updates