Model save

README.md

@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0083
-- Accuracy: 1.0
+- Loss: 1.2750
+- Accuracy: 0.5
 
 ## Model description
 
@@ -50,16 +50,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 1    | 0.0902          | 1.0      |
-| No log        | 2.0   | 2    | 0.0155          | 1.0      |
-| No log        | 3.0   | 3    | 0.0094          | 1.0      |
-| No log        | 4.0   | 4    | 0.0080          | 1.0      |
-| No log        | 5.0   | 5    | 0.0078          | 1.0      |
-| No log        | 6.0   | 6    | 0.0080          | 1.0      |
-| No log        | 7.0   | 7    | 0.0082          | 1.0      |
-| No log        | 8.0   | 8    | 0.0083          | 1.0      |
-| No log        | 9.0   | 9    | 0.0083          | 1.0      |
-| No log        | 10.0  | 10   | 0.0083          | 1.0      |
+| No log        | 1.0   | 1    | 0.1575          | 1.0      |
+| No log        | 2.0   | 2    | 0.1166          | 1.0      |
+| No log        | 3.0   | 3    | 0.2496          | 1.0      |
+| No log        | 4.0   | 4    | 0.4093          | 0.5      |
+| No log        | 5.0   | 5    | 0.6732          | 0.5      |
+| No log        | 6.0   | 6    | 0.9200          | 0.5      |
+| No log        | 7.0   | 7    | 1.0925          | 0.5      |
+| No log        | 8.0   | 8    | 1.1963          | 0.5      |
+| No log        | 9.0   | 9    | 1.2521          | 0.5      |
+| No log        | 10.0  | 10   | 1.2750          | 0.5      |
 
 
 ### Framework versions

config.json

@@ -9,15 +9,15 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "cats",
-    "1": "dogs"
+    "0": "ironed_tshirts",
+    "1": "not_ironed_tshirts"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "cats": "0",
-    "dogs": "1"
+    "ironed_tshirts": "0",
+    "not_ironed_tshirts": "1"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:10e8c9929819974ac918ad9a20d1f76b1aed3c7a2972023e70cc1e2263cd0ece
+oid sha256:09ad2c426d21893a2db10ca60448cd7ce1a5b4ea263ff92f3e4bfdfd7b09103b
 size 343223968

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5a6a4a662e50561aaa2781136164d315e2117d83e4206937a4fa45b2cce1ad30
+oid sha256:071949e542030a35e69afcefe2c7cd476752ca71fece3ae17daaed1d718b663f
 size 5240