Training in progress epoch 0

README.md

@@ -15,11 +15,11 @@ 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:
-- Train Loss: 0.3650
-- Train Accuracy: 0.9401
-- Train Top-3-accuracy: 0.9676
-- Validation Loss: 0.0202
-- Validation Accuracy: 1.0
+- Train Loss: 0.1415
+- Train Accuracy: 0.9739
+- Train Top-3-accuracy: 0.9863
+- Validation Loss: 0.0055
+- Validation Accuracy: 0.9993
 - Validation Top-3-accuracy: 1.0
 - Epoch: 0
 
@@ -40,19 +40,19 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 645, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}}, 'dynamic': True, 'initial_scale': 32768.0, 'dynamic_growth_steps': 2000}
+- optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 0.0002, 'decay_steps': 2720, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}}, 'dynamic': True, 'initial_scale': 32768.0, 'dynamic_growth_steps': 2000}
 - training_precision: mixed_float16
 
 ### Training results
 
 | Train Loss | Train Accuracy | Train Top-3-accuracy | Validation Loss | Validation Accuracy | Validation Top-3-accuracy | Epoch |
 |:----------:|:--------------:|:--------------------:|:---------------:|:-------------------:|:-------------------------:|:-----:|
-| 0.3650     | 0.9401         | 0.9676               | 0.0202          | 1.0                 | 1.0                       | 0     |
+| 0.1415     | 0.9739         | 0.9863               | 0.0055          | 0.9993              | 1.0                       | 0     |
 
 
 ### Framework versions
 
 - Transformers 4.31.0
 - TensorFlow 2.12.0
-- Datasets 2.13.1
+- Datasets 2.14.0
 - Tokenizers 0.13.3

config.json

@@ -9,37 +9,113 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "banana",
-    "1": "bell_pepper",
+    "0": "apple",
+    "1": "ball",
     "10": "cereal_box",
-    "11": "comb",
-    "12": "dry_battery",
-    "2": "ball",
-    "3": "binder",
-    "4": "bowl",
-    "5": "cell_phone",
+    "11": "coffee_mug",
+    "12": "comb",
+    "13": "dry_battery",
+    "14": "flashlight",
+    "15": "food_bag",
+    "16": "food_box",
+    "17": "food_can",
+    "18": "food_cup",
+    "19": "food_jar",
+    "2": "banana",
+    "20": "garlic",
+    "21": "glue_stick",
+    "22": "greens",
+    "23": "hand_towel",
+    "24": "instant_noodles",
+    "25": "keyboard",
+    "26": "kleenex",
+    "27": "lemon",
+    "28": "lightbulb",
+    "29": "lime",
+    "3": "bell_pepper",
+    "30": "marker",
+    "31": "mushroom",
+    "32": "notebook",
+    "33": "onion",
+    "34": "orange",
+    "35": "peach",
+    "36": "pear",
+    "37": "pitcher",
+    "38": "plate",
+    "39": "pliers",
+    "4": "binder",
+    "40": "potato",
+    "41": "rubber_eraser",
+    "42": "scissors",
+    "43": "shampoo",
+    "44": "soda_can",
+    "45": "sponge",
+    "46": "stapler",
+    "47": "tomato",
+    "48": "toothbrush",
+    "49": "toothpaste",
+    "5": "bowl",
+    "50": "water_bottle",
     "6": "calculator",
-    "7": "cap",
-    "8": "camera",
-    "9": "coffee_mug"
+    "7": "camera",
+    "8": "cap",
+    "9": "cell_phone"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "ball": "2",
-    "banana": "0",
-    "bell_pepper": "1",
-    "binder": "3",
-    "bowl": "4",
+    "apple": "0",
+    "ball": "1",
+    "banana": "2",
+    "bell_pepper": "3",
+    "binder": "4",
+    "bowl": "5",
     "calculator": "6",
-    "camera": "8",
-    "cap": "7",
-    "cell_phone": "5",
+    "camera": "7",
+    "cap": "8",
+    "cell_phone": "9",
     "cereal_box": "10",
-    "coffee_mug": "9",
-    "comb": "11",
-    "dry_battery": "12"
+    "coffee_mug": "11",
+    "comb": "12",
+    "dry_battery": "13",
+    "flashlight": "14",
+    "food_bag": "15",
+    "food_box": "16",
+    "food_can": "17",
+    "food_cup": "18",
+    "food_jar": "19",
+    "garlic": "20",
+    "glue_stick": "21",
+    "greens": "22",
+    "hand_towel": "23",
+    "instant_noodles": "24",
+    "keyboard": "25",
+    "kleenex": "26",
+    "lemon": "27",
+    "lightbulb": "28",
+    "lime": "29",
+    "marker": "30",
+    "mushroom": "31",
+    "notebook": "32",
+    "onion": "33",
+    "orange": "34",
+    "peach": "35",
+    "pear": "36",
+    "pitcher": "37",
+    "plate": "38",
+    "pliers": "39",
+    "potato": "40",
+    "rubber_eraser": "41",
+    "scissors": "42",
+    "shampoo": "43",
+    "soda_can": "44",
+    "sponge": "45",
+    "stapler": "46",
+    "tomato": "47",
+    "toothbrush": "48",
+    "toothpaste": "49",
+    "water_bottle": "50"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

logs/train/events.out.tfevents.1690226946.0dde144b55f7.2692.0.v2

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

logs/validation/events.out.tfevents.1690233702.0dde144b55f7.2692.1.v2

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

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ac8d8b81f05baea347d4a51a098f210dc3721efb1cc06ca3a001a3ee00ef1a77
-size 343503544
+oid sha256:164ba036d9807b05a74093fe6243e2b2e864ee137682ad78f180968ef0ec78c9
+size 343620280