NhatPham
/

vit-base-patch16-224-recylce-ft

Image Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

NhatPham commited on May 27, 2022

Commit

e9c0a3e

•

1 Parent(s): 5485ac7

add model

Files changed (2) hide show

config.json +9 -8
pytorch_model.bin +2 -2

config.json CHANGED Viewed

@@ -1,24 +1,25 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224",
   "architectures": [
     "ViTForImageClassification"
   ],
   "attention_probs_dropout_prob": 0.0,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "0",
-    "1": "1",
-    "2": "2"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "0": "0",
-    "1": "1",
-    "2": "2"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
@@ -28,5 +29,5 @@
   "patch_size": 16,
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.11.3"
 }

 {
+  "_name_or_path": "NhatPham/vit-base-patch16-224-recylce-ft",
   "architectures": [
     "ViTForImageClassification"
   ],
   "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
+    "0": "Object",
+    "1": "Recycle",
+    "2": "Non-Recycle "
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
+    "Non-Recycle ": 2,
+    "Object": 0,
+    "Recycle": 1
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
   "patch_size": 16,
   "qkv_bias": true,
   "torch_dtype": "float32",
+  "transformers_version": "4.19.2"
 }

pytorch_model.bin CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:219582ea825de141ea58bb976810cf1493c2d8421213438df5806159f6be9489
-size 343282929

 version https://git-lfs.github.com/spec/v1
+oid sha256:43753b62f79b09a3f760e437778302b9b9f7440e43b431f3b8e7a0a21483bc97
+size 343267249