VMamba v0 Classification checkpoints
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 | 292 | did'nt add | config | |
VMamba-S | ImageNet-1K | 224x224 | 83.5 | 44M | 238 | true | config | |
VMamba-B | ImageNet-1K | 224x224 | 83.2 | 75M | 260 | did'nt add | config | |
VMamba-B* | ImageNet-1K | 224x224 | 83.7 | 75M | 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