-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yaml
46 lines (41 loc) · 860 Bytes
/
config.yaml
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
#DATASET
DATASET:
TRAIN_LIST: "/home/jhebu/dataset/CoronaryArtery/challengedata/train.lst"
VALIDATE_LIST: "/home/jhebu/dataset/CoronaryArtery/challengedata/validate.lst"
TEST_LIST: "/home/jhebu/dataset/CoronaryArtery/challengedata/train.lst"
PATCH_SIZE: [32, 32, 32]
SAMPLE_NUMBER: 1000
#MODEL
MODEL:
NAME: "da_resnet34_3d"
DIM: "3d"
BN: "bn"
INPUT_CHANNEL: 3
NCLASS: 2
#SOLVER
SOLVER:
NAME: "Adam"
LEARN_RATE: 0.001
LR_MODE: "poly"
LR_STEPS: 40
EPOCHS: 1000000
MOMENTUM: 0.9
WEIGTH_DECAY: 0.001
#LOSS
LOSS:
TYPE: "ce_loss+dice_loss+mse_loss"
WEIGHT: [1.0, 1.0, 2.0]
CLASS_WEIGHT: [1.0, 10.0]
IGNORE_INDEX: -100
#TRAIN
TRAIN:
WORKER: 1
BATCH_SIZE: 128
#VALIDATAE
VALIDATE:
WORKER: 1
BATCH_SIZE: 128
#TEST
TEST:
WORKER: 1
BATCH_SIZE: 1