/
config_getting_started_classification_rad3d.yaml
104 lines (104 loc) · 1.94 KB
/
config_getting_started_classification_rad3d.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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
batch_size: 1
clip_grad: null
clip_mode: null
data_augmentation: {}
data_postprocessing: {}
data_postprocessing_after_reverse_one_hot_encoding: {}
data_preprocessing:
normalize: null
enable_padding: false
grid_aggregator_overlap: crop
in_memory: false
inference_mechanism:
grid_aggregator_overlap: crop
patch_overlap: 0
learning_rate: 0.001
loss_function: cel
medcam_enabled: false
metrics:
accuracy:
average: weighted
mdmc_average: samplewise
multi_class: true
subset_accuracy: false
threshold: 0.5
balanced_accuracy: null
cel: null
classification_accuracy: null
f1:
average: weighted
f1:
average: weighted
mdmc_average: samplewise
multi_class: true
threshold: 0.5
iou:
iou:
reduction: sum
reduction: elementwise_mean
threshold: 0.5
per_label_one_hot_accuracy: null
precision:
average: weighted
mdmc_average: samplewise
multi_class: true
precision:
average: weighted
threshold: 0.5
recall:
average: weighted
mdmc_average: samplewise
multi_class: true
threshold: 0.5
modality: rad
model:
amp: false
architecture: vgg16
base_filters: 32
class_list:
- 0
- 1
- 2
data_type: FP32
dimension: 3
final_layer: softmax
ignore_label_validation: null
norm_type: batch
num_channels: 1
num_classes: 3
onnx_export: false
print_summary: false
save_at_every_epoch: false
type: torch
nested_training:
testing: -5
validation: -5
num_epochs: 1
optimizer:
type: adam
parallel_compute_command: ''
patch_sampler: uniform
patch_size:
- 32
- 32
- 32
patience: 1
pin_memory_dataloader: false
print_rgb_label_warning: true
problem_type: classification
q_max_length: 1
q_num_workers: 0
q_samples_per_volume: 1
q_verbose: false
save_output: false
save_training: false
scaling_factor: 1
scheduler:
step_size: 0.0002
type: triangle
track_memory_usage: false
verbose: false
version:
maximum: 0.0.20-dev
minimum: 0.0.14
weighted_loss: true