Instructions to use darragh/swinunetr-btcv-small with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use darragh/swinunetr-btcv-small with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("darragh/swinunetr-btcv-small", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| { | |
| "architecture": "SwinUNETR", | |
| "img_size": 96, | |
| "in_channels": 1, | |
| "out_channels": 14, | |
| "depths": [ | |
| 2, | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "num_heads": [ | |
| 3, | |
| 6, | |
| 12, | |
| 24 | |
| ], | |
| "feature_size": 24, | |
| "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 | |
| } | |