Upload folder using huggingface_hub

checkpoint-1302/config.json

@@ -0,0 +1,118 @@
+{
+  "_name_or_path": "dima806/traffic_sign_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": "Bicycles crossing",
+    "1": "Children crossing",
+    "2": "Danger Ahead",
+    "3": "Dangerous curve to the left",
+    "4": "Dangerous curve to the right",
+    "5": "Dont Go Left",
+    "6": "Dont Go Left or Right",
+    "7": "Dont Go Right",
+    "8": "Dont Go straight",
+    "9": "Dont Go straight or left",
+    "10": "Dont overtake from Left",
+    "11": "Fences",
+    "12": "Go Left",
+    "13": "Go Left or right",
+    "14": "Go Right",
+    "15": "Go left or straight",
+    "16": "Go right or straight",
+    "17": "Go straight",
+    "18": "Go straight or right",
+    "19": "Heavy Vehicle Accidents",
+    "20": "Horn",
+    "21": "No Car",
+    "22": "No Uturn",
+    "23": "No entry",
+    "24": "No horn",
+    "25": "No stopping",
+    "26": "Road Divider",
+    "27": "Roundabout mandatory",
+    "28": "Speed limit (15km/h)",
+    "29": "Speed limit (30km/h)",
+    "30": "Speed limit (40km/h)",
+    "31": "Speed limit (50km/h)",
+    "32": "Speed limit (5km/h)",
+    "33": "Speed limit (60km/h)",
+    "34": "Speed limit (70km/h)",
+    "35": "Train Crossing",
+    "36": "Under Construction",
+    "37": "Unknown",
+    "38": "Uturn",
+    "39": "Zebra Crossing",
+    "40": "ZigZag Curve",
+    "41": "keep Left",
+    "42": "keep Right",
+    "43": "speed limit (80km/h)",
+    "44": "watch out for cars"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "Bicycles crossing": 0,
+    "Children crossing": 1,
+    "Danger Ahead": 2,
+    "Dangerous curve to the left": 3,
+    "Dangerous curve to the right": 4,
+    "Dont Go Left": 5,
+    "Dont Go Left or Right": 6,
+    "Dont Go Right": 7,
+    "Dont Go straight": 8,
+    "Dont Go straight or left": 9,
+    "Dont overtake from Left": 10,
+    "Fences": 11,
+    "Go Left": 12,
+    "Go Left or right": 13,
+    "Go Right": 14,
+    "Go left or straight": 15,
+    "Go right or straight": 16,
+    "Go straight": 17,
+    "Go straight or right": 18,
+    "Heavy Vehicle Accidents": 19,
+    "Horn": 20,
+    "No Car": 21,
+    "No Uturn": 22,
+    "No entry": 23,
+    "No horn": 24,
+    "No stopping": 25,
+    "Road Divider": 26,
+    "Roundabout mandatory": 27,
+    "Speed limit (15km/h)": 28,
+    "Speed limit (30km/h)": 29,
+    "Speed limit (40km/h)": 30,
+    "Speed limit (50km/h)": 31,
+    "Speed limit (5km/h)": 32,
+    "Speed limit (60km/h)": 33,
+    "Speed limit (70km/h)": 34,
+    "Train Crossing": 35,
+    "Under Construction": 36,
+    "Unknown": 37,
+    "Uturn": 38,
+    "Zebra Crossing": 39,
+    "ZigZag Curve": 40,
+    "keep Left": 41,
+    "keep Right": 42,
+    "speed limit (80km/h)": 43,
+    "watch out for cars": 44
+  },
+  "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.40.0"
+}

checkpoint-1302/model.safetensors

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

checkpoint-1302/optimizer.pt

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

checkpoint-1302/preprocessor_config.json

@@ -0,0 +1,36 @@
+{
+  "_valid_processor_keys": [
+    "images",
+    "do_resize",
+    "size",
+    "resample",
+    "do_rescale",
+    "rescale_factor",
+    "do_normalize",
+    "image_mean",
+    "image_std",
+    "return_tensors",
+    "data_format",
+    "input_data_format"
+  ],
+  "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-1302/rng_state.pth

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

checkpoint-1302/scheduler.pt

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

checkpoint-1302/trainer_state.json

@@ -0,0 +1,62 @@
+{
+  "best_metric": 2.437823534011841,
+  "best_model_checkpoint": "traffic_sign_detection/checkpoint-1302",
+  "epoch": 3.0,
+  "eval_steps": 500,
+  "global_step": 1302,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9154777345438824,
+      "eval_loss": 2.4788408279418945,
+      "eval_runtime": 94.2266,
+      "eval_samples_per_second": 98.189,
+      "eval_steps_per_second": 12.279,
+      "step": 434
+    },
+    {
+      "epoch": 1.1520737327188941,
+      "grad_norm": 1.8849073648452759,
+      "learning_rate": 3.2028753993610226e-07,
+      "loss": 2.2883,
+      "step": 500
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9212062256809338,
+      "eval_loss": 2.4474833011627197,
+      "eval_runtime": 92.6376,
+      "eval_samples_per_second": 99.873,
+      "eval_steps_per_second": 12.49,
+      "step": 868
+    },
+    {
+      "epoch": 2.3041474654377883,
+      "grad_norm": 1.7652419805526733,
+      "learning_rate": 1.2060702875399362e-07,
+      "loss": 2.2472,
+      "step": 1000
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9217466493731085,
+      "eval_loss": 2.437823534011841,
+      "eval_runtime": 127.3999,
+      "eval_samples_per_second": 72.622,
+      "eval_steps_per_second": 9.082,
+      "step": 1302
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 1302,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "total_flos": 3.2275448973660856e+18,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-1302/training_args.bin

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

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:75e4241cb2b58bcfa6edfff460ac81c35cc1d04a857e9debd62f5ab116e931cd
+oid sha256:cea1ed40c74c0524c4b67eadc44805de3ff51e7844fe68baa8a7cb34ab3591ae
 size 343356252

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5114471d5cab7938c1227fd3d6eee0294b100b9097ce4cc96d915e7722bad73d
+oid sha256:e92628c61f8ba7db2c939a24588d94ee65c82e6785199d767cb1a2d783262898
 size 4920