-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathrun_cwru.sh
46 lines (33 loc) · 928 Bytes
/
run_cwru.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
uid_head=(834)
methods=(dann dann dann dann dann dann)
datasets=cwru
models=inceptiontime
trial_num=6
sd=(12FanEndFault_0.007
12FanEndFault_0.014
12FanEndFault_0.021
12DriveEndFault_0.007
12DriveEndFault_0.014
12DriveEndFault_0.021
)
td=(12DriveEndFault_0.007
12DriveEndFault_0.014
12DriveEndFault_0.021
12FanEndFault_0.007
12FanEndFault_0.014
12FanEndFault_0.021
)
totalStep=48000
gpu_array=( 1 1 2 2 3 3)
for ((j=0;j<${trial_num};j++))
do
let uid_array[$j]=uid_head[0]+$[$j]
done
for ((i=0;i<${trial_num};i++))
do
nohup python3 main.py \
--logdir=logs --modeldir=models \
--method=${methods[$i]} --model=${models} --dataset=${datasets} --sources=${sd[$i]} \
--target=${td[$i]} --uid=${uid_array[$i]} --moving_average=True --gpu_index=${gpu_array[$i]} --debugnum=0 --steps=${totalStep} --gpumem=7168 >./logs/cwru_uid_${uid_array[$i]}.txt 2>&1 &
done
echo ${uid_array[*]}