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: custom_dataset_augmented
-      type: imagefolder
-      config: default
-      split: train
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.6065573770491803
-    - name: F1
-      type: f1
-      value: 0.3511904761904762
+  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 custom_dataset_augmented dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.9847
-- Accuracy: 0.6066
-- F1: 0.3512
+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,38 +10,14 @@
   "hidden_size": 768,
   "id2label": {
     "0": "aeroplane",
-    "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"
+    "1": "car"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
     "aeroplane": "0",
-    "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"
+    "car": "1"
   },
   "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:156a2a0c5abbe677a5d26824b6069f51313a41eeaa0067fbfd0662b23499b718
-size 343302829
+oid sha256:e65f55480ef978eafaaf888e46b7779f01b893e4a7250b958b765230ee3c8313
+size 343265965

training_args.bin

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