-
Notifications
You must be signed in to change notification settings - Fork 0
/
max_loss.sh
38 lines (32 loc) · 1.14 KB
/
max_loss.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
# for seed in 1 2 3 4 5 6 7 8 9 10
# do
# for mem_size in 100;
# do
# for kernel_epoch in 1 2 5 10 20 40 70 100 200 250 500 1000 2000;
# do
# for act in "softmax" "sparsemax"
# do
# for data in "mnist" "cifar10"
# do
# CUDA_VISIBLE_DEVICES=5 python3 memory_retrieval_max_loss.py --memory_size $mem_size --kernel_epoch $kernel_epoch --activation $act --data $data --mode "UMHN" --seed $seed --rerun 44
# done
# done
# done
# done
# done
for seed in 1 2 3 4 5 6 7 8 9 10
do
for mem_size in 100;
do
for kernel_epoch in 1 2 5 10 20 40 70 100 200 250 500 1000 2000;
do
for act in "softmax" "sparsemax"
do
for data in "mnist" "cifar10"
do
CUDA_VISIBLE_DEVICES=5 python3 memory_retrieval_max_loss.py --memory_size $mem_size --kernel_epoch $kernel_epoch --activation $act --data $data --mode "UMHN" --seed $seed --rerun 44
done
done
done
done
done