tensorgirl
/

TFaugvit

Image Classification

Inference Endpoints

Model card Files Files and versions Community

tensorgirl commited on Sep 20, 2023

Commit

d86ba11

·

1 Parent(s): fd82cc1

Update config.json

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -11,10 +11,10 @@
   "dropout": 0.1,
   "emb_dropout": 0.1,
   "heads": 16,
-  "image_size": 32,
   "mlp_dim": 256,
   "model_type": "augvit",
-  "num_classes": 10,
-  "patch_size": 4,
   "transformers_version": "4.33.2"
 }

   "dropout": 0.1,
   "emb_dropout": 0.1,
   "heads": 16,
+  "image_size": 224,
   "mlp_dim": 256,
   "model_type": "augvit",
+  "num_classes": 1000,
+  "patch_size": 32,
   "transformers_version": "4.33.2"
 }