Skip to content

VMamba v0 Classification checkpoints

Compare
Choose a tag to compare
@MzeroMiko MzeroMiko released this 18 Feb 03:28
· 48 commits to main since this release

Checkpoints for VMamba (alias of vssm version 0)

These checkpoints correspond to the experiments done before date #20240119.

name pretrain resolution acc@1 #params FLOPs best epoch use ema config
VMamba-T ImageNet-1K 224x224 82.2 22M 4.5G 5.6G 292 did'nt add config
VMamba-S ImageNet-1K 224x224 83.5 44M 9.1G 11.2G 238 true config
VMamba-B ImageNet-1K 224x224 83.2 75M 15.2G 18.0G 260 did'nt add config
VMamba-B* ImageNet-1K 224x224 83.7 75M 15.2G 18.0G 241 true config

Most backbone models trained without ema, which do not enhance performance \cite(Swin-Transformer). We use ema because our model is still under development, without hyperparameter tuning.

The checkpoints used in object detection and segmentation is VMamba-B with droppath 0.5 + no ema. VMamba-B* represents for VMamba-B with droppath 0.6 + ema, the performance of which is non-ema: 83.3 in epoch 262; ema: 83.7 in epoch 241