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 +3 -4
config.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"architecture": "vit_base_patch16_224
|
| 3 |
"architectures": [
|
| 4 |
"ViTForImageClassification"
|
| 5 |
],
|
|
@@ -30,11 +30,10 @@
|
|
| 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 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"architecture": "vit_base_patch16_224",
|
| 3 |
"architectures": [
|
| 4 |
"ViTForImageClassification"
|
| 5 |
],
|
|
|
|
| 30 |
"transformers_version": "4.57.6",
|
| 31 |
|
| 32 |
"pretrained_cfg": {
|
| 33 |
+
"architecture": "vit_base_patch16_224",
|
| 34 |
+
"file": "timm_model.bin",
|
| 35 |
"num_classes": 1,
|
| 36 |
"input_size": [3, 224, 224],
|
|
|
|
|
|
|
| 37 |
"mean": [0.5, 0.5, 0.5],
|
| 38 |
"std": [0.5, 0.5, 0.5]
|
| 39 |
}
|