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
Update config.json
Browse files- config.json +13 -2
config.json
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
{
|
|
|
|
| 2 |
"architectures": [
|
| 3 |
"ViTForImageClassification"
|
| 4 |
],
|
|
@@ -26,5 +27,15 @@
|
|
| 26 |
"pooler_act": "tanh",
|
| 27 |
"pooler_output_size": 768,
|
| 28 |
"qkv_bias": true,
|
| 29 |
-
"transformers_version": "4.57.6"
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
{
|
| 2 |
+
"architecture": "vit_base_patch16_224",
|
| 3 |
"architectures": [
|
| 4 |
"ViTForImageClassification"
|
| 5 |
],
|
|
|
|
| 27 |
"pooler_act": "tanh",
|
| 28 |
"pooler_output_size": 768,
|
| 29 |
"qkv_bias": true,
|
| 30 |
+
"transformers_version": "4.57.6",
|
| 31 |
+
|
| 32 |
+
"pretrained_cfg": {
|
| 33 |
+
"architecture": "vit_base_patch16_224",
|
| 34 |
+
"num_classes": 1,
|
| 35 |
+
"input_size": [3, 224, 224],
|
| 36 |
+
"pool_size": null,
|
| 37 |
+
"crop_pct": 0.9,
|
| 38 |
+
"mean": [0.5, 0.5, 0.5],
|
| 39 |
+
"std": [0.5, 0.5, 0.5]
|
| 40 |
+
}
|
| 41 |
+
}
|