Upload folder using huggingface_hub

checkpoint-930/config.json

@@ -0,0 +1,38 @@
+{
+  "_name_or_path": "dima806/wildfire_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": "Both_smoke_and_fire",
+    "1": "Fire_confounding_elements",
+    "2": "Forested_areas_without_confounding_elements",
+    "3": "Smoke_confounding_elements",
+    "4": "Smoke_from_fires"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "Both_smoke_and_fire": 0,
+    "Fire_confounding_elements": 1,
+    "Forested_areas_without_confounding_elements": 2,
+    "Smoke_confounding_elements": 3,
+    "Smoke_from_fires": 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.38.2"
+}

checkpoint-930/model.safetensors

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

checkpoint-930/optimizer.pt

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

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

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

checkpoint-930/scheduler.pt

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

checkpoint-930/trainer_state.json

@@ -0,0 +1,73 @@
+{
+  "best_metric": 0.41760945320129395,
+  "best_model_checkpoint": "wildfire_types_image_detection/checkpoint-930",
+  "epoch": 5.0,
+  "eval_steps": 500,
+  "global_step": 930,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.8825847123719465,
+      "eval_loss": 0.4752463698387146,
+      "eval_runtime": 347.2919,
+      "eval_samples_per_second": 3.654,
+      "eval_steps_per_second": 0.458,
+      "step": 186
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.8959810874704491,
+      "eval_loss": 0.4484263062477112,
+      "eval_runtime": 347.5981,
+      "eval_samples_per_second": 3.651,
+      "eval_steps_per_second": 0.457,
+      "step": 372
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 5.007945537567139,
+      "learning_rate": 1.465909090909091e-06,
+      "loss": 0.3867,
+      "step": 500
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.8983451536643026,
+      "eval_loss": 0.43289777636528015,
+      "eval_runtime": 344.1545,
+      "eval_samples_per_second": 3.687,
+      "eval_steps_per_second": 0.462,
+      "step": 558
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9014972419227738,
+      "eval_loss": 0.4210251271724701,
+      "eval_runtime": 343.269,
+      "eval_samples_per_second": 3.697,
+      "eval_steps_per_second": 0.463,
+      "step": 744
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9030732860520094,
+      "eval_loss": 0.41760945320129395,
+      "eval_runtime": 347.7036,
+      "eval_samples_per_second": 3.65,
+      "eval_steps_per_second": 0.457,
+      "step": 930
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 930,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "total_flos": 1.1472997540690022e+18,
+  "train_batch_size": 16,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-930/training_args.bin

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

config.json

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

model.safetensors

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7d409e635e141ad0ed6084f7f575a0394c77507c7ffa50c02b229ee29e4ebecc
-size 4091
+oid sha256:f0ee8eef683969b12870780a79dc2936e9eaad5f9efaec3460b56de7f6a7f435
+size 4411