|
{ |
|
"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 |
|
} |
|
} |