forked from FT-ZHOU-ZZZ/CMTA
-
Notifications
You must be signed in to change notification settings - Fork 1
/
run.sh
65 lines (60 loc) · 3.48 KB
/
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
CUDA_VISIBLE_DEVICES=0 python main.py --which_splits 5foldcv \
--dataset tcga_blca \
--data_root_dir /master/zhou_feng_tao/data/TCGA/BLCA/x20 \
--modal coattn \
--model cmta \
--num_epoch 30 \
--batch_size 1 \
--loss nll_surv_l1 \
--lr 0.001 \
--optimizer SGD \
--scheduler None \
--alpha 1.0
CUDA_VISIBLE_DEVICES=0 python main.py --which_splits 5foldcv \
--dataset tcga_brca \
--data_root_dir /master/zhou_feng_tao/data/TCGA/BRCA/x20 \
--modal coattn \
--model cmta \
--num_epoch 30 \
--batch_size 1 \
--loss nll_surv_l1 \
--lr 0.001 \
--optimizer SGD \
--scheduler None \
--alpha 100.0
CUDA_VISIBLE_DEVICES=0 python main.py --which_splits 5foldcv \
--dataset tcga_gbmlgg \
--data_root_dir /master/zhou_feng_tao/data/TCGA/GBMLGG/x20 \
--modal coattn \
--model cmta \
--num_epoch 30 \
--batch_size 1 \
--loss nll_surv_l1 \
--lr 0.001 \
--optimizer SGD \
--scheduler None \
--alpha 1.0
CUDA_VISIBLE_DEVICES=0 python main.py --which_splits 5foldcv \
--dataset tcga_luad \
--data_root_dir /master/zhou_feng_tao/data/TCGA/LUAD/x20 \
--modal coattn \
--model cmta \
--num_epoch 30 \
--batch_size 1 \
--loss nll_surv_l1 \
--lr 0.001 \
--optimizer SGD \
--scheduler None \
--alpha 0.0001
CUDA_VISIBLE_DEVICES=0 python main.py --which_splits 5foldcv \
--dataset tcga_ucec \
--data_root_dir /master/zhou_feng_tao/data/TCGA/UCEC/x20 \
--modal coattn \
--model cmta \
--num_epoch 30 \
--batch_size 1 \
--loss nll_surv_l1 \
--lr 0.001 \
--optimizer SGD \
--scheduler None \
--alpha 1.0