-
Notifications
You must be signed in to change notification settings - Fork 1
/
config_prediction.yaml
79 lines (63 loc) · 1.48 KB
/
config_prediction.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
wandb:
project_name: test
project_team: tail-upenn
online: false
project:
device: gpu
num_workers: 12
num_epochs: 3
pin_memory: True
load_model: True
val_interval: 2
time: 0
epoch: 0
hyperparameters:
batch_size: 8
weights: True
optimizer_fn: adam
adam_learning_rate: .0003
sgd_learning_rate: .01
sgd_momentum: 0.9
sgd_weight_decay: .0001
sgd_nesterov: True
loss_fn: crossentropy
multiplier: [1,1,1,1]
tversky_alpha: .5
tversky_beta: .5
focal_alpha: 1
focal_gamma: 2
scheduler_fn: warm
scheduler_patience: 5
earlystopping: True
earlystop_patience: 10
model:
max_layer_size: 1024
min_layer_size: 64
calibration:
custom: True
image:
phantom_format: dcm
mask_format: png
image_height: 600
image_width: 360
image_channels: 1
mask_channels: 1
mask_labels: 4
labels:
0: background
1: low_risk
2: moderate_risk
3: high_risk
load:
path: data/checkpoints/20230502_215115_best_checkpoint.pth.tar
paths:
train_img_dir: data/train/phantom/
train_mask_dir: data/train/mask/
val_img_dir: data/val/phantom/
val_mask_dir: data/val/mask/
test_img_dir: data/test/phantom/
test_mask_dir: data/test/mask/
predictions_dir: data/predictions/
submissions_dir: data/submissions/
calibrations_dir: data/calibrations/
confidences_dir: data/confidences/