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
Add model
Browse files- config.json +0 -3
config.json
CHANGED
|
@@ -3,9 +3,6 @@
|
|
| 3 |
"num_classes": 1,
|
| 4 |
"num_features": 768,
|
| 5 |
"global_pool": "token",
|
| 6 |
-
"label_names": [
|
| 7 |
-
"your_single_class_label"
|
| 8 |
-
],
|
| 9 |
"pretrained_cfg": {
|
| 10 |
"tag": "augreg2_in21k_ft_in1k",
|
| 11 |
"custom_load": false,
|
|
|
|
| 3 |
"num_classes": 1,
|
| 4 |
"num_features": 768,
|
| 5 |
"global_pool": "token",
|
|
|
|
|
|
|
|
|
|
| 6 |
"pretrained_cfg": {
|
| 7 |
"tag": "augreg2_in21k_ft_in1k",
|
| 8 |
"custom_load": false,
|