Upload folder using huggingface_hub

checkpoint-94/config.json

@@ -0,0 +1,52 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "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-94/model.safetensors

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

checkpoint-94/optimizer.pt

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

checkpoint-94/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-94/rng_state.pth

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

checkpoint-94/scheduler.pt

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

checkpoint-94/trainer_state.json

@@ -0,0 +1,43 @@
+{
+  "best_metric": 2.4650940895080566,
+  "best_model_checkpoint": "garbage_types_image_detection/checkpoint-94",
+  "epoch": 1.0,
+  "eval_steps": 500,
+  "global_step": 94,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.1185,
+      "eval_loss": 2.4650940895080566,
+      "eval_model_preparation_time": 0.005,
+      "eval_runtime": 38.603,
+      "eval_samples_per_second": 103.619,
+      "eval_steps_per_second": 3.238,
+      "step": 94
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 94,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "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": 4.64993609859072e+17,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-94/training_args.bin

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

config.json

@@ -0,0 +1,52 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "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"
+}

model.safetensors

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

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
+  }
+}

training_args.bin

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