-
Notifications
You must be signed in to change notification settings - Fork 37
/
Copy pathtest_one_example.sh
59 lines (52 loc) · 1.94 KB
/
test_one_example.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
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
device=0
# LOG=${save_dir}"res.log"
# echo ${LOG}
# depth=(9)
# n_ctx=(12)
# t_n_ctx=(4)
# for i in "${!depth[@]}";do
# for j in "${!n_ctx[@]}";do
# ## train on the VisA dataset
# base_dir=${depth[i]}_${n_ctx[j]}_${t_n_ctx[0]}_multiscale
# save_dir=./checkpoints_memory/${base_dir}/
# CUDA_VISIBLE_DEVICES=${device} python test.py --dataset mvtec \
# --data_path /root/data/mvdataset --save_path ./results/${base_dir}/zero_shot \
# --checkpoint_path ${save_dir}epoch_15.pth \
# --features_list 6 12 18 24 --image_size 518 --depth ${depth[i]} --n_ctx ${n_ctx[j]} --t_n_ctx ${t_n_ctx[0]}
# wait
# done
# done
# LOG=${save_dir}"res.log"
# echo ${LOG}
# depth=(9)
# n_ctx=(12)
# t_n_ctx=(4)
# for i in "${!depth[@]}";do
# for j in "${!n_ctx[@]}";do
# ## train on the VisA dataset
# base_dir=${depth[i]}_${n_ctx[j]}_${t_n_ctx[0]}_multiscale_visa
# save_dir=./checkpoints/${base_dir}/
# CUDA_VISIBLE_DEVICES=${device} python test.py --dataset visa \
# --data_path /remote-home/iot_zhouqihang/data/Visa --save_path ./results/${base_dir}/zero_shot \
# --checkpoint_path ${save_dir}epoch_15.pth \
# --features_list 6 12 18 24 --image_size 518 --depth ${depth[i]} --n_ctx ${n_ctx[j]} --t_n_ctx ${t_n_ctx[0]}
# wait
# done
# done
LOG=${save_dir}"res.log"
echo ${LOG}
depth=(9)
n_ctx=(12)
t_n_ctx=(4)
for i in "${!depth[@]}";do
for j in "${!n_ctx[@]}";do
## train on the VisA dataset
base_dir=${depth[i]}_${n_ctx[j]}_${t_n_ctx[0]}_multiscale
save_dir=./checkpoints_mul/${base_dir}/
CUDA_VISIBLE_DEVICES=${device} python test_one_example.py \
--image_path /remote-home/iot_zhouqihang/old_root/data/mvdataset/carpet/test/hole/004.png \
--checkpoint_path ${save_dir}epoch_15.pth \
--features_list 6 12 18 24 --image_size 518 --depth ${depth[i]} --n_ctx ${n_ctx[j]} --t_n_ctx ${t_n_ctx[0]}
wait
done
done