Anwarkh1
/

Skin_Cancer-Image_Classification

Image Classification

Inference Endpoints

Model card Files Files and versions Community

Anwarkh1 commited on Mar 17

Commit

fe0ef07

•

1 Parent(s): 8f5fc75

Upload ViTForImageClassification

Files changed (2) hide show

config.json +11 -12
model.safetensors +2 -2

config.json CHANGED Viewed

@@ -1,24 +1,23 @@
 {
   "architectures": [
-    "BertModel"
   ],
-  "attention_probs_dropout_prob": 0.1,
-  "classifier_dropout": null,
   "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "layer_norm_eps": 1e-12,
-  "max_position_embeddings": 512,
-  "model_type": "bert",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
-  "pad_token_id": 0,
-  "position_embedding_type": "absolute",
   "torch_dtype": "float32",
-  "transformers_version": "4.38.2",
-  "type_vocab_size": 2,
-  "use_cache": true,
-  "vocab_size": 30522
 }

 {
+  "_name_or_path": "google/vit-base-patch16-224-in21k",
   "architectures": [
+    "ViTForImageClassification"
   ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
   "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
+  "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "layer_norm_eps": 1e-12,
+  "model_type": "vit",
   "num_attention_heads": 12,
+  "num_channels": 3,
   "num_hidden_layers": 12,
+  "patch_size": 16,
+  "qkv_bias": true,
   "torch_dtype": "float32",
+  "transformers_version": "4.38.2"
 }

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7a1ec3b6b6a34ba97d4bd850f58ec0101c5af37dd68c6bbb96a40da67eefc2fd
-size 437951328

 version https://git-lfs.github.com/spec/v1
+oid sha256:467eadb4eaab7fd9393b0c0da29b369bc47890ce839e9200393d563e65df58f4
+size 343239356