-
Notifications
You must be signed in to change notification settings - Fork 36
/
eval.sh
24 lines (18 loc) · 1.07 KB
/
eval.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
if [ ! $1 ];
then
DATA_DIR=/path/to/imagenet/val
else
DATA_DIR="$1"
fi
if [ ! $2 ];
then
MODEL_DIR=/path/to/checkpoint
else
MODEL_DIR="$2"
fi
python3 validate.py $DATA_DIR --model lvvit_s --checkpoint $MODEL_DIR/lvvit_s-26M-224-83.3.pth.tar --no-test-pool --amp --img-size 224 -b 64
# python3 validate.py $DATA_DIR --model lvvit_m --checkpoint $MODEL_DIR/lvvit_m-56M-224-84.0.pth.tar --no-test-pool --amp --img-size 224 -b 64
# python3 validate.py $DATA_DIR --model lvvit_l --checkpoint $MODEL_DIR/lvvit_l-150M-448-86.2.pth.tar --no-test-pool --amp --img-size 448 -b 64
# python3 validate.py $DATA_DIR --model lvvit_s --checkpoint $MODEL_DIR/lvvit_s-26M-224-83.3.pth.tar --no-test-pool --amp --img-size 224 -b 64 --real-labels real.json
# python3 validate.py $DATA_DIR --model lvvit_m --checkpoint $MODEL_DIR/lvvit_m-56M-224-84.0.pth.tar --no-test-pool --amp --img-size 224 -b 64 --real-labels real.json
# python3 validate.py $DATA_DIR --model lvvit_l --checkpoint $MODEL_DIR/lvvit_l-150M-448-86.2.pth.tar --no-test-pool --amp --img-size 448 -b 64 --real-labels real.json