Upload folder using huggingface_hub

checkpoint-6000/config.json

@@ -0,0 +1,52 @@
+{
+  "_name_or_path": "dima806/garbage_types_image_detection",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "battery",
+    "1": "biological",
+    "2": "brown-glass",
+    "3": "cardboard",
+    "4": "clothes",
+    "5": "green-glass",
+    "6": "metal",
+    "7": "paper",
+    "8": "plastic",
+    "9": "shoes",
+    "10": "trash",
+    "11": "white-glass"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "battery": 0,
+    "biological": 1,
+    "brown-glass": 2,
+    "cardboard": 3,
+    "clothes": 4,
+    "green-glass": 5,
+    "metal": 6,
+    "paper": 7,
+    "plastic": 8,
+    "shoes": 9,
+    "trash": 10,
+    "white-glass": 11
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.44.0"
+}

checkpoint-6000/model.safetensors

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

checkpoint-6000/optimizer.pt

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

checkpoint-6000/preprocessor_config.json

@@ -0,0 +1,22 @@
+{
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

checkpoint-6000/rng_state.pth

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

checkpoint-6000/scheduler.pt

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

checkpoint-6000/trainer_state.json

@@ -0,0 +1,217 @@
+{
+  "best_metric": 1.0335067510604858,
+  "best_model_checkpoint": "garbage_types_image_detection/checkpoint-6000",
+  "epoch": 10.0,
+  "eval_steps": 500,
+  "global_step": 6000,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.8333333333333334,
+      "grad_norm": 1.107461929321289,
+      "learning_rate": 9.243697478991597e-07,
+      "loss": 2.3762,
+      "step": 500
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.6217918622848201,
+      "eval_loss": 2.261021137237549,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 230.6368,
+      "eval_samples_per_second": 110.824,
+      "eval_steps_per_second": 3.464,
+      "step": 600
+    },
+    {
+      "epoch": 1.6666666666666665,
+      "grad_norm": 1.2842817306518555,
+      "learning_rate": 8.403361344537815e-07,
+      "loss": 2.1389,
+      "step": 1000
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.84358372456964,
+      "eval_loss": 1.9786875247955322,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 231.1495,
+      "eval_samples_per_second": 110.578,
+      "eval_steps_per_second": 3.457,
+      "step": 1200
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 1.3137625455856323,
+      "learning_rate": 7.563025210084033e-07,
+      "loss": 1.871,
+      "step": 1500
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.8843896713615024,
+      "eval_loss": 1.6969841718673706,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 229.6174,
+      "eval_samples_per_second": 111.316,
+      "eval_steps_per_second": 3.48,
+      "step": 1800
+    },
+    {
+      "epoch": 3.3333333333333335,
+      "grad_norm": 1.404272198677063,
+      "learning_rate": 6.722689075630252e-07,
+      "loss": 1.62,
+      "step": 2000
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.8964006259780908,
+      "eval_loss": 1.4749500751495361,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 229.017,
+      "eval_samples_per_second": 111.607,
+      "eval_steps_per_second": 3.489,
+      "step": 2400
+    },
+    {
+      "epoch": 4.166666666666667,
+      "grad_norm": 1.3687832355499268,
+      "learning_rate": 5.88235294117647e-07,
+      "loss": 1.4226,
+      "step": 2500
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 4.501117706298828,
+      "learning_rate": 5.042016806722689e-07,
+      "loss": 1.272,
+      "step": 3000
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9044992175273865,
+      "eval_loss": 1.3128881454467773,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 227.3834,
+      "eval_samples_per_second": 112.409,
+      "eval_steps_per_second": 3.514,
+      "step": 3000
+    },
+    {
+      "epoch": 5.833333333333333,
+      "grad_norm": 1.4435608386993408,
+      "learning_rate": 4.2016806722689076e-07,
+      "loss": 1.1605,
+      "step": 3500
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.9131846635367762,
+      "eval_loss": 1.2000573873519897,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 229.3389,
+      "eval_samples_per_second": 111.451,
+      "eval_steps_per_second": 3.484,
+      "step": 3600
+    },
+    {
+      "epoch": 6.666666666666667,
+      "grad_norm": 1.3652788400650024,
+      "learning_rate": 3.361344537815126e-07,
+      "loss": 1.0793,
+      "step": 4000
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.9211658841940532,
+      "eval_loss": 1.1223890781402588,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 228.3054,
+      "eval_samples_per_second": 111.955,
+      "eval_steps_per_second": 3.5,
+      "step": 4200
+    },
+    {
+      "epoch": 7.5,
+      "grad_norm": 1.9194189310073853,
+      "learning_rate": 2.5210084033613445e-07,
+      "loss": 1.0186,
+      "step": 4500
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.9250782472613458,
+      "eval_loss": 1.0719075202941895,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 228.4622,
+      "eval_samples_per_second": 111.878,
+      "eval_steps_per_second": 3.497,
+      "step": 4800
+    },
+    {
+      "epoch": 8.333333333333334,
+      "grad_norm": 1.7687724828720093,
+      "learning_rate": 1.680672268907563e-07,
+      "loss": 0.9775,
+      "step": 5000
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.9281298904538341,
+      "eval_loss": 1.0426900386810303,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 233.5981,
+      "eval_samples_per_second": 109.419,
+      "eval_steps_per_second": 3.42,
+      "step": 5400
+    },
+    {
+      "epoch": 9.166666666666666,
+      "grad_norm": 1.763899564743042,
+      "learning_rate": 8.403361344537815e-08,
+      "loss": 0.9562,
+      "step": 5500
+    },
+    {
+      "epoch": 10.0,
+      "grad_norm": 3.2325987815856934,
+      "learning_rate": 0.0,
+      "loss": 0.9383,
+      "step": 6000
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.928169014084507,
+      "eval_loss": 1.0335067510604858,
+      "eval_model_preparation_time": 0.0049,
+      "eval_runtime": 233.555,
+      "eval_samples_per_second": 109.439,
+      "eval_steps_per_second": 3.421,
+      "step": 6000
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 6000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 10,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 2.97130916699947e+19,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-6000/training_args.bin

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

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "dima806/garbage_types_image_detection",
   "architectures": [
     "ViTForImageClassification"
   ],

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6dd8fe7131b97c1a831dda21343aa9c31cf015568ad90a080bc55a1193d58352
+oid sha256:b9d9dec6fe465c468dd9f4cd986e633c5ad770d2a5603c22a74cfeaf6c8327b4
 size 343254736

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f29d01e98b321d6bc09b19af4a6fa0a48a4bb82d4b942c03e76fd51e739844ef
+oid sha256:bbc0c982884ddde220b2d119f9fb47fdaddccb089e8ef201a0cf5a3daaebc02b
 size 5112