SwinUnetr / config.json
zllhhhh's picture
Update config.json
ac24875 verified
raw
history blame
No virus
693 Bytes
{
"vision_config": {
"_name_or_path": "",
"architecture": "Swin3DClass",
"image_size": 96,
"in_channels": 1,
"out_channels": 14,
"depths": [
2,
2,
2,
2
],
"num_heads": [
3,
6,
12,
24
],
"hidden_size": 768,
"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,
"projection_dim" : 768,
"top_k": 49,
"top_p": 1.0
},
"vision_config_dict": {
"hidden_size": 768,
"intermediate_size": 4096,
"projection_dim": 768
}
}