Upload folder using huggingface_hub

checkpoint-2793/config.json

@@ -0,0 +1,38 @@
+{
+  "_name_or_path": "dima806/vessel_ship_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": "Cargo",
+    "1": "Carrier",
+    "2": "Cruise",
+    "3": "Military",
+    "4": "Tankers"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "Cargo": 0,
+    "Carrier": 1,
+    "Cruise": 2,
+    "Military": 3,
+    "Tankers": 4
+  },
+  "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.36.1"
+}

checkpoint-2793/model.safetensors

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

checkpoint-2793/optimizer.pt

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

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

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

checkpoint-2793/scheduler.pt

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

checkpoint-2793/trainer_state.json

@@ -0,0 +1,240 @@
+{
+  "best_metric": 0.11951557546854019,
+  "best_model_checkpoint": "vessel_ship_types_image_detection/checkpoint-2793",
+  "epoch": 21.0,
+  "eval_steps": 500,
+  "global_step": 2793,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9849056603773585,
+      "eval_loss": 0.15518580377101898,
+      "eval_runtime": 18.1982,
+      "eval_samples_per_second": 116.495,
+      "eval_steps_per_second": 3.682,
+      "step": 133
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9844339622641509,
+      "eval_loss": 0.15392659604549408,
+      "eval_runtime": 18.1823,
+      "eval_samples_per_second": 116.597,
+      "eval_steps_per_second": 3.685,
+      "step": 266
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9849056603773585,
+      "eval_loss": 0.14783743023872375,
+      "eval_runtime": 18.2262,
+      "eval_samples_per_second": 116.316,
+      "eval_steps_per_second": 3.676,
+      "step": 399
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.2652990264255911e-06,
+      "loss": 0.1176,
+      "step": 500
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9863207547169811,
+      "eval_loss": 0.14441388845443726,
+      "eval_runtime": 18.2843,
+      "eval_samples_per_second": 115.946,
+      "eval_steps_per_second": 3.664,
+      "step": 532
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9844339622641509,
+      "eval_loss": 0.14357155561447144,
+      "eval_runtime": 18.1412,
+      "eval_samples_per_second": 116.861,
+      "eval_steps_per_second": 3.693,
+      "step": 665
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.9863207547169811,
+      "eval_loss": 0.13874255120754242,
+      "eval_runtime": 17.9915,
+      "eval_samples_per_second": 117.834,
+      "eval_steps_per_second": 3.724,
+      "step": 798
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.9849056603773585,
+      "eval_loss": 0.13630421459674835,
+      "eval_runtime": 18.3986,
+      "eval_samples_per_second": 115.226,
+      "eval_steps_per_second": 3.642,
+      "step": 931
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 1.00452016689847e-06,
+      "loss": 0.1015,
+      "step": 1000
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.9858490566037735,
+      "eval_loss": 0.13331656157970428,
+      "eval_runtime": 18.2197,
+      "eval_samples_per_second": 116.358,
+      "eval_steps_per_second": 3.677,
+      "step": 1064
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.9839622641509433,
+      "eval_loss": 0.13424086570739746,
+      "eval_runtime": 18.2024,
+      "eval_samples_per_second": 116.468,
+      "eval_steps_per_second": 3.681,
+      "step": 1197
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.9853773584905661,
+      "eval_loss": 0.13191621005535126,
+      "eval_runtime": 18.3372,
+      "eval_samples_per_second": 115.612,
+      "eval_steps_per_second": 3.654,
+      "step": 1330
+    },
+    {
+      "epoch": 11.0,
+      "eval_accuracy": 0.9858490566037735,
+      "eval_loss": 0.13006728887557983,
+      "eval_runtime": 18.386,
+      "eval_samples_per_second": 115.305,
+      "eval_steps_per_second": 3.644,
+      "step": 1463
+    },
+    {
+      "epoch": 11.28,
+      "learning_rate": 7.437413073713492e-07,
+      "loss": 0.0919,
+      "step": 1500
+    },
+    {
+      "epoch": 12.0,
+      "eval_accuracy": 0.9863207547169811,
+      "eval_loss": 0.12602895498275757,
+      "eval_runtime": 18.4226,
+      "eval_samples_per_second": 115.076,
+      "eval_steps_per_second": 3.637,
+      "step": 1596
+    },
+    {
+      "epoch": 13.0,
+      "eval_accuracy": 0.9867924528301887,
+      "eval_loss": 0.1253497451543808,
+      "eval_runtime": 18.1742,
+      "eval_samples_per_second": 116.649,
+      "eval_steps_per_second": 3.687,
+      "step": 1729
+    },
+    {
+      "epoch": 14.0,
+      "eval_accuracy": 0.9858490566037735,
+      "eval_loss": 0.12341459840536118,
+      "eval_runtime": 18.2588,
+      "eval_samples_per_second": 116.108,
+      "eval_steps_per_second": 3.669,
+      "step": 1862
+    },
+    {
+      "epoch": 15.0,
+      "eval_accuracy": 0.9863207547169811,
+      "eval_loss": 0.12174314260482788,
+      "eval_runtime": 18.285,
+      "eval_samples_per_second": 115.942,
+      "eval_steps_per_second": 3.664,
+      "step": 1995
+    },
+    {
+      "epoch": 15.04,
+      "learning_rate": 4.829624478442281e-07,
+      "loss": 0.0842,
+      "step": 2000
+    },
+    {
+      "epoch": 16.0,
+      "eval_accuracy": 0.9853773584905661,
+      "eval_loss": 0.12054015696048737,
+      "eval_runtime": 18.2322,
+      "eval_samples_per_second": 116.278,
+      "eval_steps_per_second": 3.675,
+      "step": 2128
+    },
+    {
+      "epoch": 17.0,
+      "eval_accuracy": 0.9844339622641509,
+      "eval_loss": 0.12279084324836731,
+      "eval_runtime": 18.3759,
+      "eval_samples_per_second": 115.369,
+      "eval_steps_per_second": 3.646,
+      "step": 2261
+    },
+    {
+      "epoch": 18.0,
+      "eval_accuracy": 0.9858490566037735,
+      "eval_loss": 0.12070772796869278,
+      "eval_runtime": 18.6962,
+      "eval_samples_per_second": 113.392,
+      "eval_steps_per_second": 3.584,
+      "step": 2394
+    },
+    {
+      "epoch": 18.8,
+      "learning_rate": 2.2218358831710707e-07,
+      "loss": 0.078,
+      "step": 2500
+    },
+    {
+      "epoch": 19.0,
+      "eval_accuracy": 0.9853773584905661,
+      "eval_loss": 0.12023171037435532,
+      "eval_runtime": 18.284,
+      "eval_samples_per_second": 115.948,
+      "eval_steps_per_second": 3.664,
+      "step": 2527
+    },
+    {
+      "epoch": 20.0,
+      "eval_accuracy": 0.9853773584905661,
+      "eval_loss": 0.1197926327586174,
+      "eval_runtime": 18.2605,
+      "eval_samples_per_second": 116.098,
+      "eval_steps_per_second": 3.669,
+      "step": 2660
+    },
+    {
+      "epoch": 21.0,
+      "eval_accuracy": 0.9853773584905661,
+      "eval_loss": 0.11951557546854019,
+      "eval_runtime": 18.2193,
+      "eval_samples_per_second": 116.36,
+      "eval_steps_per_second": 3.677,
+      "step": 2793
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 2926,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 22,
+  "save_steps": 500,
+  "total_flos": 1.3800144559581757e+19,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-2793/training_args.bin

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

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "dima806/vessel_ship_types_image_detection",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -34,5 +34,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.35.2"
+  "transformers_version": "4.36.1"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:30f3155419517d01e2859d32e6e36d669af32d3c2ef16a1124c44238cc0675b0
+oid sha256:9d6892f03c45be58b3ee4a3f9a034f8c42f0c276909fd85914b36cf5dde5e5bd
 size 343233204

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8b47ff0c38f70b9f70a6e1842f49d6c65e70286aafea8e7452b04937f8184af9
-size 4091
+oid sha256:315d35a1a3f2a0a472013e1a65ab38cbecc1388338ec0819aeb19a3a020db568
+size 4219