Skip to content

GitLab

Commit b1ce2942 authored by luopl's avatar luopl
Browse files

Update multidcu_validate.sh

parent dd9dddf2
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mambavision/multidcu_validate.sh
View file @ b1ce2942
...@@ -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 \
--num-gpu 2 --num-gpu 2
\ No newline at end of file
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