Spaces:
Sleeping
Sleeping
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 |