Skip to content

Latest commit

 

History

History
14 lines (11 loc) · 1.4 KB

README.md

File metadata and controls

14 lines (11 loc) · 1.4 KB

Model Weights and Configurations

Model #param. Top-1 Acc. Config Log
FastVim-T.ckpt 7M 75.4 FastVimT.yaml FastVimT.csv
FastVim-S.ckpt 26M 81.1 FastVimS.yaml FastVimS.csv
FastVim-B.ckpt 98M 82.6 FastVimB.yaml FastVimB.csv
Vim-B w/ LN.ckpt 98M 82.6 VimB.yaml VimB.csv

Notes:

  • For reproducibility, make sure overall batch size remains 1024 across GPUs/Nodes.
  • trainer/global_step in log files refers to gradient steps with batch size 1024.
  • Modify imagenet_train_dir_path and imagenet_val_dir_path flags in datasets_supervised.py code.