tumor-segmentation-v1 / segment_params.json
jamesdolezal's picture
Upload 2 files
666fa9d
{
"params": {
"arch": "FPN",
"encoder_name": "resnet34",
"in_channels": 3,
"out_classes": 1
},
"mpp": 10,
"size": 2048,
"train_batch_size": 4,
"val_batch_size": 16,
"epochs": 10
}