Spaces:
Sleeping
Sleeping
File size: 366 Bytes
1cc0005 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
device: 'cuda'
data:
dataset: "/home/tumor/LungTumorSegmentation/Resources/Inference/malvik_inference_data2.yaml"
scale_dim:
d_0: 256
d_1: 256
d_2: 128
model:
architecture: UNet_test
weights: "/home/tumor/model_last_128.pth"
loss: DiceLoss
filter_base: 128
filter_expansion: 2
inference:
save_directory: '/home/tumor/data/SoftLabels/'
|