Instructions to use RationAI/vit-patch16-224-prostate with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use RationAI/vit-patch16-224-prostate with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("RationAI/vit-patch16-224-prostate", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Create timm_config.json
Browse files- timm_config.json +9 -0
timm_config.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architecture": "vit_base_patch16_224",
|
| 3 |
+
"num_classes": 1,
|
| 4 |
+
"input_size": [3, 224, 224],
|
| 5 |
+
"pool_size": null,
|
| 6 |
+
"crop_pct": 0.9,
|
| 7 |
+
"mean": [0.5, 0.5, 0.5],
|
| 8 |
+
"std": [0.5, 0.5, 0.5]
|
| 9 |
+
}
|