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