validate.sh 232 Bytes
Newer Older
luopl's avatar
init  
luopl committed
1
2
3
4
5
#!/bin/bash
DATA_PATH="/ImageNet/val"
BS=128
checkpoint='/model_weights/mambavision_tiny_1k.pth.tar'

luopl's avatar
luopl committed
6
python validate.py --model mamba_vision_T --checkpoint=$checkpoint --data-dir=$DATA_PATH --batch-size $BS --input-size 3 224 224
luopl's avatar
init  
luopl committed
7