-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.sh
executable file
·27 lines (25 loc) · 1021 Bytes
/
run.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
#!/bin/bash
dataset_list='MVTec VisA'
MVTec_list='bottle cable capsule carpet grid hazelnut leather metal nut pill screw tile toothbrush transistor wood zipper'
VisA_list='candle, capsules, cashew, chewinggum, fryum, macaroni1, macaroni2, pcb1, pcb2, pcb3, pcb4, pipe_fryum'
for dataset in $dataset_list
do
if [ "$dataset" == 'MVTec' ]
then
for class in $MVTec_list
do
# Modify the below line based on your requirements and system setup
echo "Dataset: $dataset, Class: $class"
accelerate launch main.py --yaml_config "./configs/$dataset/$class.yaml"
done
elif [ "$dataset" == 'VisA' ]
then
IFS=', ' read -r -a visA_classes <<< "$VisA_list"
for class in "${visA_classes[@]}"
do
# Modify the below line based on your requirements and system setup
echo "Dataset: $dataset, Class: $class"
accelerate launch main.py --yaml_config "./configs/$dataset/$class.yaml"
done
fi
done