{ | |
"architecture": "SwinUNETR", | |
"img_size": 96, | |
"in_channels": 1, | |
"out_channels": 14, | |
"depths": [ | |
2, | |
2, | |
2, | |
2 | |
], | |
"num_heads": [ | |
3, | |
6, | |
12, | |
24 | |
], | |
"feature_size": 12, | |
"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 | |
} |