Vemund Fredriksen
Add training pipeline (#21)
1cc0005 unverified
raw
history blame
No virus
545 Bytes
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