Spaces:
Sleeping
Sleeping
File size: 739 Bytes
1cc0005 |
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 |
device: 'cuda'
data:
train_dataset: ["D:\\Repos\\LungTumorSegmentation\\Resources\\Test\\data_test_train.yaml",
"D:\\Repos\\LungTumorSegmentation\\Resources\\Test\\data_test_train.yaml",
"D:\\Repos\\LungTumorSegmentation\\Resources\\Test\\data_test_train.yaml"]
aug_prob: 0.0
scale_dim:
d_0: 64
d_1: 64
d_2: 64
model:
architecture: UNet_monai
loss: DiceLoss
weights:
filter_base: 32
filter_expansion: 2
filters: [2, 3, 4, 5]
optimizer:
name: 'Adam'
lr: 0.001
train:
max_epochs: 3
val_frequency: 1
output_threshold: 0.5
metric_path: "D:\\Repos\\LungTumorSegmentation\\metrics.csv"
save_frequency: 5
model_directory: "D:\\Repos\\LungTumorSegmentation\\models\\"
batch_size: 3 |