swinunetr-btcv-base / config.json
darragh's picture
Upload model files
dde995f
{
"architecture": "SwinUNETR",
"img_size": 96,
"in_channels": 1,
"out_channels": 14,
"depths": [
2,
2,
2,
2
],
"num_heads": [
3,
6,
12,
24
],
"feature_size": 48,
"norm_name": "instance",
"drop_rate": 0.0,
"attn_drop_rate": 0.0,
"dropout_path_rate": 0.0,
"normalize": true,
"use_checkpoint": false,
"spatial_dims": 3
}