Spaces:
Sleeping
Sleeping
File size: 451 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 |
device: 'cuda'
data:
train_dataset: ""
aug_prob: 0.1
scale_dim:
d_0: 128
d_1: 128
d_2: 128
model:
architecture: UNet4
loss: DiceLoss
filter_base: 16
filter_expansion: 2
weights:
optimizer:
name: 'Adam'
lr: 0.000025
train:
max_epochs: 500
val_frequency: 1
output_threshold: 0.5
metric_path: "/home/tumor/out/metric.csv"
save_frequency: 50
model_directory: "/home/tumor/out/models/"
batch_size: 10
|