Spaces:
Sleeping
Sleeping
device: 'cuda' | |
data: | |
train_dataset: "/cluster/work/sosevle/LungTumorSegmentation/Resources/Idun_Train_2/data.yaml" | |
aug_prob: 0.1 | |
scale_dim: | |
d_0: 256 | |
d_1: 256 | |
d_2: 128 | |
model: | |
architecture: UNet4 | |
loss: DiceLoss | |
weights: | |
filter_base: 32 | |
filter_expansion: 2 | |
optimizer: | |
name: 'Adam' | |
lr: 0.0001 | |
train: | |
max_epochs: 5000 | |
val_frequency: 1 | |
output_threshold: 0.5 | |
metric_path: "/cluster/work/sosevle/metrics/t32.csv" | |
save_frequency: 5 | |
model_directory: "/cluster/work/sosevle/models/32/" | |
batch_size: 10 |