-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathloop_jobs.sh
executable file
·38 lines (31 loc) · 1.13 KB
/
loop_jobs.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
#!/bin/bash
data_path='./data/'
file_out='logs_3' # Output log file
act='leaky_relu'
for task in 'pMNIST' # Choose among the tasks ['MNIST','pMNIST','PTB','PTB_5','copying','copyingVariable']
do
for seq_len in 784 # For MNIST set seq_len to 784
do
for m in 32 # Number of reflections
do
for hidden_units in 32
do
for model_type in 'oRNN'
do
for lr in 0.001 # Learning rate
do
for seed in 1 # Random seed
do
for bs in 5 # Batch size
do
python train.py $seed $task $seq_len $model_type $hidden_units $m $lr $bs $act $data_path &> $file_out &
# python test.py $seed $task $seq_len $model_type $hidden_units $m $lr $bs $act $data_path &> $file_out &
# sbatch job.sh $seed $task $seq_len $model_type $hidden_units $m $lr $bs $act $data_path
done
done
done
done
done
done
done
done