Skip to content

GitLab

Commit dd9dddf2 authored by luopl's avatar luopl
Browse files

Update validate.sh

parent 2eefe3d6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mambavision/validate.sh
View file @ dd9dddf2
...@@ -3,5 +3,5 @@ DATA_PATH="/ImageNet/val" ...@@ -3,5 +3,5 @@ DATA_PATH="/ImageNet/val"
BS=128 BS=128
checkpoint='/model_weights/mambavision_tiny_1k.pth.tar' checkpoint='/model_weights/mambavision_tiny_1k.pth.tar'
python validate.py --model mamba_vision_T --checkpoint=$checkpoint --data_dir=$DATA_PATH --batch-size $BS --input-size 3 224 224 python validate.py --model mamba_vision_T --checkpoint=$checkpoint --data-dir=$DATA_PATH --batch-size $BS --input-size 3 224 224
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment