-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest.sh
23 lines (22 loc) · 1.02 KB
/
test.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# offical evaluation. use to evaluate the valset.
python -m torch.distributed.launch --nproc_per_node=1 run_beit3_finetuning.py \
--model beit3_large_patch16_224 \
--task imagenet \
--batch_size 128 \
--sentencepiece_model ./pretrained_model/beit3.spm \
--finetune ./pretrained_model/beit3_large_itc_patch16_224.pth \
--data_path ./data/kitchen \
--eval \
--dist_eval \
--checkpoint ./saved_model/checkpoint-best-saved/mp_rank_00_model_states.pt
# hwj's rebuilt evaluation. use to evaluate the testset.
python -m torch.distributed.launch --nproc_per_node=1 run_beit3_finetuning.py \
--model beit3_large_patch16_224 \
--task imagenet \
--batch_size 128 \
--sentencepiece_model ./pretrained_model/beit3.spm \
--finetune ./pretrained_model/beit3_large_itc_patch16_224.pth \
--data_path ./data/kitchen \
--hwj_test \
--dist_eval \
--checkpoint ./saved_model/checkpoint-best-saved/mp_rank_00_model_states.pt