Model save

README.md

@@ -1,30 +1,12 @@
 ---
 base_model: d071696/vit-finetune-scrap
 tags:
-- image-classification
-- image-feature-extraction
-- image-to-text
 - generated_from_trainer
 datasets:
-- arrow
-metrics:
-- accuracy
+- imagefolder
 model-index:
 - name: vit-finetune-scrap
-  results:
-  - task:
-      name: Image Classification
-      type: image-classification
-    dataset:
-      name: d071696/scraps1
-      type: arrow
-      config: default
-      split: train
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.9963782696177063
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,10 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-finetune-scrap
 
-This model is a fine-tuned version of [d071696/vit-finetune-scrap](https://huggingface.co/d071696/vit-finetune-scrap) on the d071696/scraps1 dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.0129
-- Accuracy: 0.9964
+This model is a fine-tuned version of [d071696/vit-finetune-scrap](https://huggingface.co/d071696/vit-finetune-scrap) on the imagefolder dataset.
 
 ## Model description
 
@@ -64,9 +43,6 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.0035        | 3.22  | 1000 | 0.2143          | 0.9486   |
 
 
 ### Framework versions

config.json

@@ -9,33 +9,35 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "aluminium",
-    "1": "batteries",
-    "10": "takeaway cups",
-    "2": "cardboard",
-    "3": "disposable plates",
-    "4": "glass",
-    "5": "hard plastic",
-    "6": "paper",
-    "7": "paper towel",
-    "8": "polystyrene",
-    "9": "soft plastics"
+    "0": "carton boxes",
+    "1": "old newspapers",
+    "10": "used juice box",
+    "11": "used plastic bottles labels",
+    "2": "old newspapers with plastic bottles and plastic garbage",
+    "3": "paper boxes and plastic trash bags",
+    "4": "paper waste",
+    "5": "plastic bottles",
+    "6": "plastic bottles and plastic bags with paper boxes",
+    "7": "plastic bottles and plastic garbage",
+    "8": "plastic garbage",
+    "9": "plastic garbage and paper garbage"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "aluminium": "0",
-    "batteries": "1",
-    "cardboard": "2",
-    "disposable plates": "3",
-    "glass": "4",
-    "hard plastic": "5",
-    "paper": "6",
-    "paper towel": "7",
-    "polystyrene": "8",
-    "soft plastics": "9",
-    "takeaway cups": "10"
+    "carton boxes": "0",
+    "old newspapers": "1",
+    "old newspapers with plastic bottles and plastic garbage": "2",
+    "paper boxes and plastic trash bags": "3",
+    "paper waste": "4",
+    "plastic bottles": "5",
+    "plastic bottles and plastic bags with paper boxes": "6",
+    "plastic bottles and plastic garbage": "7",
+    "plastic garbage": "8",
+    "plastic garbage and paper garbage": "9",
+    "used juice box": "10",
+    "used plastic bottles labels": "11"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d563541c04dbb7251d614a78ad4b90a6173da7438f314d9f54f7354ebee3ff11
-size 343251660
+oid sha256:9b61cb4e03c9bfcb4d59e04874ddb0628ad12043bcb2d77be6e778249886b256
+size 343254736

runs/Mar29_18-42-22_X5C922065N/events.out.tfevents.1711734147.X5C922065N.53009.6

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4810560ff653082ad5673c08aa8290117a11a73694a1a6c9ae6c5bdf632333cc
+size 6106

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:617efafadeca5d575973508bc1fb23b178b0f91ade0be63e4197fe64ab5c4ac7
+oid sha256:72eb25d1bd046d95141d050a7046c0b70c3fb9f17a91b9b38a15dbfa48b5b07c
 size 4920