Skip to content

Commit 5c426ae

Browse files
committed
sm2
Signed-off-by: myron <[email protected]>
1 parent c19e171 commit 5c426ae

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

auto3dseg/tasks/kits23/input.yaml

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# KiTS23 Auto3dseg minimal user input
1+
# KiTS23 Auto3DSeg user input
22

33
modality: CT # primary modality
44
dataroot: /data/kits23 # dataset location
@@ -8,20 +8,20 @@ class_names: # names for tensorboard, and label index gro
88
- { name: mass, index: [2,3] }
99
- { name: tumor, index: [2] }
1010

11+
# OPTIONAL
1112
# use final sigmoid activation (instead of the default softmax), since KiTS regions are overlapping (multi-label segmentation)
1213
# this is optional to set, the system auto-detects overlapping labels automatically.
1314
sigmoid: true
1415

15-
1616
# the config below is optional, but it explicitly sets params as it was used during KiTS23 challenge
17-
# otherwise, the defaults are (auto_scale_allowed is True) and the system will attempt to guess these settings according to the available GPU (e.g. make batch size larger)
17+
# otherwise, the defaults are used, auto_scale_allowed is True and the system will attempt to guess these settings according to the available GPU (e.g. make batch size larger)
1818
auto_scale_allowed: false
19-
roi_size: [256, 256, 256]
2019
num_epochs: 600
21-
loss: {_target_: DiceLoss}
20+
resample: true
21+
resample_resolution: [0.78125, 0.78125, 0.78125]
22+
# roi_size: [256, 256, 256]
23+
# roi_size: [336, 336, 336]
24+
# loss: {_target_: DiceLoss}
2225
# batch_size: 1
23-
# resample: true
24-
# resample_resolution: [0.78125, 0.78125, 0.78125] # this is auto determined, uncomment to set another value manually
25-
26-
26+
# augment_mode: ct_ax_1
2727

0 commit comments

Comments
 (0)