README.md

January 28, 2025 ยท View on GitHub

Model Weights and Configurations

Model#param.Top-1 Acc.ConfigLog
FastVim-T.ckpt7M75.4FastVimT.yamlFastVimT.csv
FastVim-S.ckpt26M81.1FastVimS.yamlFastVimS.csv
FastVim-B.ckpt98M82.6FastVimB.yamlFastVimB.csv
Vim-B w/ LN.ckpt98M82.6VimB.yamlVimB.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.