-
Notifications
You must be signed in to change notification settings - Fork 0
/
temp-4.sh
49 lines (43 loc) · 961 Bytes
/
temp-4.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
45
46
47
48
# python train.py \
# --dataset cifar10 \
# --model resnet32 \
# --schedule-steps 80 120 \
# --epochs 160 \
# --loss cross_entropy
# python train.py \
# --dataset cifar10 \
# --model resnet32 \
# --schedule-steps 80 120 \
# --epochs 160 \
# --loss NLL+MDCA --beta 1.0
export CUDA_VISIBLE_DEVICES="2,3"
python train.py \
--dataset cifar10 \
--model resnet32 \
--schedule-steps 80 120 \
--epochs 160 \
--loss NLL+MDCA --beta 5.0
python train.py \
--dataset cifar10 \
--model resnet32 \
--schedule-steps 80 120 \
--epochs 160 \
--loss NLL+MDCA --beta 10.0
python train.py \
--dataset cifar10 \
--model resnet32 \
--schedule-steps 80 120 \
--epochs 160 \
--loss NLL+MDCA --beta 15.0
# python train.py \
# --dataset cifar10 \
# --model resnet32 \
# --schedule-steps 80 120 \
# --epochs 160 \
# --loss NLL+MDCA --beta 20.0
# python train.py \
# --dataset cifar10 \
# --model resnet32 \
# --schedule-steps 80 120 \
# --epochs 160 \
# --loss NLL+MDCA --beta 25.0