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 | |