RationAI
/

vit-patch16-224-prostate

Model card Files Files and versions

matejpekar commited on Jun 11

Commit

392bd2f

·

verified ·

1 Parent(s): a4352cb

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "architecture": "vit_base_patch16_224",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -30,7 +30,7 @@
   "transformers_version": "4.57.6",
   "pretrained_cfg": {
-    "architecture": "vit_base_patch16_224",
     "num_classes": 1,
     "input_size": [3, 224, 224],
     "pool_size": null,

 {
+  "architecture": "vit_base_patch16_224.augreg_in21k",
   "architectures": [
     "ViTForImageClassification"
   ],
   "transformers_version": "4.57.6",
   "pretrained_cfg": {
+    "architecture": "vit_base_patch16_224.augreg_in21k",
     "num_classes": 1,
     "input_size": [3, 224, 224],
     "pool_size": null,