Model save

README.md

@@ -2,32 +2,12 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224
 tags:
-- image-classification
 - generated_from_trainer
 datasets:
 - imagefolder
-metrics:
-- accuracy
-- f1
 model-index:
 - name: fine-tuned-augmented
-  results:
-  - task:
-      name: Image Classification
-      type: image-classification
-    dataset:
-      name: test_train
-      type: imagefolder
-      config: default
-      split: test
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 1.0
-    - name: F1
-      type: f1
-      value: 1.0
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -35,11 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # fine-tuned-augmented
 
-This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the test_train dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.0120
-- Accuracy: 1.0
-- F1: 1.0
+This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
 
 ## Model description
 

config.json

@@ -10,14 +10,38 @@
   "hidden_size": 768,
   "id2label": {
     "0": "aeroplane",
-    "1": "car"
+    "1": "car",
+    "10": "people",
+    "11": "river",
+    "12": "ski",
+    "13": "snow",
+    "2": "deer",
+    "3": "dogsledge",
+    "4": "factory",
+    "5": "fireman",
+    "6": "firetruck",
+    "7": "ladder",
+    "8": "lake",
+    "9": "mountain"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
     "aeroplane": "0",
-    "car": "1"
+    "car": "1",
+    "deer": "2",
+    "dogsledge": "3",
+    "factory": "4",
+    "fireman": "5",
+    "firetruck": "6",
+    "ladder": "7",
+    "lake": "8",
+    "mountain": "9",
+    "people": "10",
+    "river": "11",
+    "ski": "12",
+    "snow": "13"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e65f55480ef978eafaaf888e46b7779f01b893e4a7250b958b765230ee3c8313
-size 343265965
+oid sha256:0b4dd47e2da49775ec42a0b004e84fcf2b71d7332e87a1cfdd7625e1105e985d
+size 343302829

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8827f23a2dec6c73e0df9872b84c30cbf6baf54081725e7bb36a24b817acda06
+oid sha256:fdb5f39138743865b65db6735f2d5171fa22d7f68017b75eb27d548bc4402554
 size 4091