Model save

README.md

@@ -15,14 +15,14 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.1693
-- eval_precision: 0.9500
-- eval_recall: 0.9484
-- eval_accuracy: 0.9484
-- eval_f1score: 0.9482
-- eval_runtime: 45.7316
-- eval_samples_per_second: 16.531
-- eval_steps_per_second: 0.525
+- eval_loss: 0.1685
+- eval_precision: 0.9495
+- eval_recall: 0.9497
+- eval_accuracy: 0.9497
+- eval_f1score: 0.9494
+- eval_runtime: 40.9327
+- eval_samples_per_second: 18.469
+- eval_steps_per_second: 0.586
 - epoch: 5.0
 - step: 950
 

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b63630614b17796fa0c928b0794da306033a335f459ae98d8445d23bcc33d3fe
+oid sha256:92acbf1630bb50244bf0a41d2fd0f8be4fb27ec8347033eaa169017970fb6487
 size 378042076

runs/May22_06-52-46_Shravani/events.out.tfevents.1716340980.Shravani.27548.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:86b67f009753185f610364f8d9156a72e0de4ef833540e710357620d83aa19c2
-size 5363
+oid sha256:c2876c68e41799c4896cdc50b7d36a04227dc3bacf302d81e57de8331c01f173
+size 9783

tmp-checkpoint-950/config.json

@@ -23,7 +23,7 @@
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
-  "num_attention_heads": 12,
+  "num_attention_heads": 8,
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 64,

tmp-checkpoint-950/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c6a65b77379eca120d8e3d90989bf2e410faf837999c9266cbd24823b86c5706
+oid sha256:92acbf1630bb50244bf0a41d2fd0f8be4fb27ec8347033eaa169017970fb6487
 size 378042076

tmp-checkpoint-950/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1edea7193a8d5e1476b5dfbf448d7aeb2d9816c98c12781d68a4e193bcf89c78
+oid sha256:45ee97705bf337fefea766f9c5c3262075d533aa8853a0aeedb5041afa419129
 size 756199354

tmp-checkpoint-950/trainer_state.json

@@ -1,6 +1,6 @@
 {
-  "best_metric": 0.1471855193376541,
-  "best_model_checkpoint": "Shravanig/vit-fire-detection\\checkpoint-570",
+  "best_metric": 0.14938032627105713,
+  "best_model_checkpoint": "Shravanig/vit-fire-detection\\checkpoint-760",
   "epoch": 5.0,
   "eval_steps": 500,
   "global_step": 950,
@@ -11,175 +11,175 @@
     {
       "epoch": 0.26,
       "learning_rate": 0.0001,
-      "loss": 0.9828,
+      "loss": 0.9987,
       "step": 50
     },
     {
       "epoch": 0.53,
       "learning_rate": 0.0002,
-      "loss": 0.6565,
+      "loss": 0.6843,
       "step": 100
     },
     {
       "epoch": 0.79,
       "learning_rate": 0.00019444444444444446,
-      "loss": 0.4455,
+      "loss": 0.5138,
       "step": 150
     },
     {
       "epoch": 1.0,
       "eval_accuracy": 0.9113756613756614,
-      "eval_f1score": 0.909933626964354,
-      "eval_loss": 0.2449590563774109,
-      "eval_precision": 0.911608601120207,
+      "eval_f1score": 0.9107285323666766,
+      "eval_loss": 0.23306569457054138,
+      "eval_precision": 0.9103883996879023,
       "eval_recall": 0.9113756613756614,
-      "eval_runtime": 55.8003,
-      "eval_samples_per_second": 13.548,
-      "eval_steps_per_second": 0.43,
+      "eval_runtime": 42.0285,
+      "eval_samples_per_second": 17.988,
+      "eval_steps_per_second": 0.571,
       "step": 190
     },
     {
       "epoch": 1.05,
       "learning_rate": 0.00018888888888888888,
-      "loss": 0.3878,
+      "loss": 0.4285,
       "step": 200
     },
     {
       "epoch": 1.32,
       "learning_rate": 0.00018333333333333334,
-      "loss": 0.4043,
+      "loss": 0.4068,
       "step": 250
     },
     {
       "epoch": 1.58,
       "learning_rate": 0.00017777777777777779,
-      "loss": 0.3663,
+      "loss": 0.3365,
       "step": 300
     },
     {
       "epoch": 1.84,
       "learning_rate": 0.00017222222222222224,
-      "loss": 0.3198,
+      "loss": 0.3161,
       "step": 350
     },
     {
       "epoch": 2.0,
-      "eval_accuracy": 0.9431216931216931,
-      "eval_f1score": 0.9430234733391375,
-      "eval_loss": 0.15777449309825897,
-      "eval_precision": 0.9432524221092315,
-      "eval_recall": 0.9431216931216931,
-      "eval_runtime": 58.5248,
-      "eval_samples_per_second": 12.918,
-      "eval_steps_per_second": 0.41,
+      "eval_accuracy": 0.8981481481481481,
+      "eval_f1score": 0.8968320885202303,
+      "eval_loss": 0.22744037210941315,
+      "eval_precision": 0.9110491043143017,
+      "eval_recall": 0.8981481481481481,
+      "eval_runtime": 41.3214,
+      "eval_samples_per_second": 18.296,
+      "eval_steps_per_second": 0.581,
       "step": 380
     },
     {
       "epoch": 2.11,
       "learning_rate": 0.0001666666666666667,
-      "loss": 0.2777,
+      "loss": 0.2911,
       "step": 400
     },
     {
       "epoch": 2.37,
       "learning_rate": 0.0001611111111111111,
-      "loss": 0.3112,
+      "loss": 0.3162,
       "step": 450
     },
     {
       "epoch": 2.63,
       "learning_rate": 0.00015555555555555556,
-      "loss": 0.2555,
+      "loss": 0.275,
       "step": 500
     },
     {
       "epoch": 2.89,
       "learning_rate": 0.00015000000000000001,
-      "loss": 0.2768,
+      "loss": 0.2686,
       "step": 550
     },
     {
       "epoch": 3.0,
-      "eval_accuracy": 0.9470899470899471,
-      "eval_f1score": 0.9469968822663881,
-      "eval_loss": 0.1471855193376541,
-      "eval_precision": 0.9475131567523525,
-      "eval_recall": 0.9470899470899471,
-      "eval_runtime": 32.47,
-      "eval_samples_per_second": 23.283,
-      "eval_steps_per_second": 0.739,
+      "eval_accuracy": 0.9378306878306878,
+      "eval_f1score": 0.937487918832369,
+      "eval_loss": 0.16444851458072662,
+      "eval_precision": 0.9376014472102792,
+      "eval_recall": 0.9378306878306878,
+      "eval_runtime": 41.7491,
+      "eval_samples_per_second": 18.108,
+      "eval_steps_per_second": 0.575,
       "step": 570
     },
     {
       "epoch": 3.16,
       "learning_rate": 0.00014444444444444444,
-      "loss": 0.235,
+      "loss": 0.2547,
       "step": 600
     },
     {
       "epoch": 3.42,
       "learning_rate": 0.0001388888888888889,
-      "loss": 0.2715,
+      "loss": 0.2721,
       "step": 650
     },
     {
       "epoch": 3.68,
       "learning_rate": 0.00013333333333333334,
-      "loss": 0.2444,
+      "loss": 0.2396,
       "step": 700
     },
     {
       "epoch": 3.95,
       "learning_rate": 0.00012777777777777776,
-      "loss": 0.2283,
+      "loss": 0.2335,
       "step": 750
     },
     {
       "epoch": 4.0,
-      "eval_accuracy": 0.9431216931216931,
-      "eval_f1score": 0.9432838391066171,
-      "eval_loss": 0.15458014607429504,
-      "eval_precision": 0.9448779078440028,
-      "eval_recall": 0.9431216931216931,
-      "eval_runtime": 59.6685,
-      "eval_samples_per_second": 12.67,
-      "eval_steps_per_second": 0.402,
+      "eval_accuracy": 0.951058201058201,
+      "eval_f1score": 0.9510381343653914,
+      "eval_loss": 0.14938032627105713,
+      "eval_precision": 0.9534914361001318,
+      "eval_recall": 0.951058201058201,
+      "eval_runtime": 45.6331,
+      "eval_samples_per_second": 16.567,
+      "eval_steps_per_second": 0.526,
       "step": 760
     },
     {
       "epoch": 4.21,
       "learning_rate": 0.00012222222222222224,
-      "loss": 0.2014,
+      "loss": 0.2412,
       "step": 800
     },
     {
       "epoch": 4.47,
       "learning_rate": 0.00011666666666666668,
-      "loss": 0.2085,
+      "loss": 0.2092,
       "step": 850
     },
     {
       "epoch": 4.74,
       "learning_rate": 0.00011111111111111112,
-      "loss": 0.2231,
+      "loss": 0.2179,
       "step": 900
     },
     {
       "epoch": 5.0,
       "learning_rate": 0.00010555555555555557,
-      "loss": 0.201,
+      "loss": 0.2058,
       "step": 950
     },
     {
       "epoch": 5.0,
-      "eval_accuracy": 0.9484126984126984,
-      "eval_f1score": 0.9482271168351205,
-      "eval_loss": 0.169330894947052,
-      "eval_precision": 0.9499665185939696,
-      "eval_recall": 0.9484126984126984,
-      "eval_runtime": 45.7316,
-      "eval_samples_per_second": 16.531,
-      "eval_steps_per_second": 0.525,
+      "eval_accuracy": 0.9497354497354498,
+      "eval_f1score": 0.9494114362396765,
+      "eval_loss": 0.1685035079717636,
+      "eval_precision": 0.9494850222122949,
+      "eval_recall": 0.9497354497354498,
+      "eval_runtime": 40.9327,
+      "eval_samples_per_second": 18.469,
+      "eval_steps_per_second": 0.586,
       "step": 950
     }
   ],

tmp-checkpoint-950/training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f4794393befeb5f69568a4ecfefe1735cfba604c435f1d5a8a1261f6b843fc7f
+oid sha256:270b5928caae40c5bca0e7cca3827e413320f661f75cdabd732e310d4ec2bcd3
 size 4792