Upload folder using huggingface_hub

checkpoint-2895/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-2895/model.safetensors

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

checkpoint-2895/optimizer.pt

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

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

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

checkpoint-2895/scheduler.pt

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

checkpoint-2895/trainer_state.json

@@ -0,0 +1,101 @@
+{
+  "best_metric": 2.523940324783325,
+  "best_model_checkpoint": "traffic_sign_detection/checkpoint-2895",
+  "epoch": 5.0,
+  "eval_steps": 500,
+  "global_step": 2895,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.8635578583765112,
+      "grad_norm": 1.785918951034546,
+      "learning_rate": 8.418277680140598e-07,
+      "loss": 2.6578,
+      "step": 500
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.849978383052313,
+      "eval_loss": 2.7859294414520264,
+      "eval_runtime": 47.349,
+      "eval_samples_per_second": 97.7,
+      "eval_steps_per_second": 12.228,
+      "step": 579
+    },
+    {
+      "epoch": 1.7271157167530224,
+      "grad_norm": 1.761311411857605,
+      "learning_rate": 6.660808435852372e-07,
+      "loss": 2.5366,
+      "step": 1000
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.8769995676610463,
+      "eval_loss": 2.669792413711548,
+      "eval_runtime": 47.4363,
+      "eval_samples_per_second": 97.52,
+      "eval_steps_per_second": 12.206,
+      "step": 1158
+    },
+    {
+      "epoch": 2.5906735751295336,
+      "grad_norm": 1.8694429397583008,
+      "learning_rate": 4.903339191564148e-07,
+      "loss": 2.4436,
+      "step": 1500
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9005620406398617,
+      "eval_loss": 2.5879104137420654,
+      "eval_runtime": 47.3175,
+      "eval_samples_per_second": 97.765,
+      "eval_steps_per_second": 12.236,
+      "step": 1737
+    },
+    {
+      "epoch": 3.454231433506045,
+      "grad_norm": 1.893329381942749,
+      "learning_rate": 3.1458699472759226e-07,
+      "loss": 2.3753,
+      "step": 2000
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9051015996541288,
+      "eval_loss": 2.5401721000671387,
+      "eval_runtime": 48.1616,
+      "eval_samples_per_second": 96.052,
+      "eval_steps_per_second": 12.022,
+      "step": 2316
+    },
+    {
+      "epoch": 4.317789291882556,
+      "grad_norm": 1.803151249885559,
+      "learning_rate": 1.3884007029876976e-07,
+      "loss": 2.3365,
+      "step": 2500
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9055339386078686,
+      "eval_loss": 2.523940324783325,
+      "eval_runtime": 47.5865,
+      "eval_samples_per_second": 97.212,
+      "eval_steps_per_second": 12.167,
+      "step": 2895
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 2895,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "total_flos": 7.172321994146857e+18,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-2895/training_args.bin

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

config.json

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

model.safetensors

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

training_args.bin

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