MRISegment_model / config.json
Nikitapyscript's picture
Upload config.json with huggingface_hub
ef864d5 verified
{
"model_type": "monai_unet_3d",
"architecture": {
"spatial_dims": 3,
"in_channels": 1,
"out_channels": 2,
"channels": [
16,
32,
64,
128,
256,
512
],
"strides": [
2,
2,
2,
2,
2,
2
],
"num_res_units": 2,
"dropout": 0.2
},
"training": {
"patch_size": [
256,
256,
256
],
"loss_function": "DiceLoss",
"optimizer": "Adam",
"learning_rate": 0.0001,
"weight_decay": 1e-05
},
"preprocessing": {
"intensity_range": [
-1000,
4000
],
"output_range": [
0,
1
],
"normalization": "ScaleIntensityRanged"
}
}