Model save

all_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 18.0,
-    "total_flos": 1.1159446583771136e+19,
-    "train_loss": 0.0,
-    "train_runtime": 27.7933,
-    "train_samples_per_second": 4317.59,
-    "train_steps_per_second": 67.462
+    "epoch": 19.0,
+    "total_flos": 1.1779415838425088e+19,
+    "train_loss": 2.718197358281989e-05,
+    "train_runtime": 389.5228,
+    "train_samples_per_second": 390.221,
+    "train_steps_per_second": 6.097
 }

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 18.0,
-    "total_flos": 1.1159446583771136e+19,
-    "train_loss": 0.0,
-    "train_runtime": 27.7933,
-    "train_samples_per_second": 4317.59,
-    "train_steps_per_second": 67.462
+    "epoch": 19.0,
+    "total_flos": 1.1779415838425088e+19,
+    "train_loss": 2.718197358281989e-05,
+    "train_runtime": 389.5228,
+    "train_samples_per_second": 390.221,
+    "train_steps_per_second": 6.097
 }

trainer_state.json

@@ -1,9 +1,9 @@
 {
   "best_metric": 0.29358136653900146,
   "best_model_checkpoint": "./drive/Shareddrives/CS198-Drones/training_output/vit-base-patch16-224_rice-leaf-disease-augmented-v2_fft/checkpoint-750",
-  "epoch": 18.0,
+  "epoch": 19.0,
   "eval_steps": 500,
-  "global_step": 2250,
+  "global_step": 2375,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -295,6 +295,31 @@
       "eval_samples_per_second": 30.349,
       "eval_steps_per_second": 0.542,
       "step": 2250
+    },
+    {
+      "epoch": 19.0,
+      "grad_norm": 0.01798320934176445,
+      "learning_rate": 0.0,
+      "loss": 0.0005,
+      "step": 2375
+    },
+    {
+      "epoch": 19.0,
+      "eval_accuracy": 0.9226190476190477,
+      "eval_loss": 0.36210647225379944,
+      "eval_runtime": 21.033,
+      "eval_samples_per_second": 15.975,
+      "eval_steps_per_second": 0.285,
+      "step": 2375
+    },
+    {
+      "epoch": 19.0,
+      "step": 2375,
+      "total_flos": 1.1779415838425088e+19,
+      "train_loss": 2.718197358281989e-05,
+      "train_runtime": 389.5228,
+      "train_samples_per_second": 390.221,
+      "train_steps_per_second": 6.097
     }
   ],
   "logging_steps": 500,
@@ -309,12 +334,12 @@
         "should_evaluate": false,
         "should_log": false,
         "should_save": true,
-        "should_training_stop": false
+        "should_training_stop": true
       },
       "attributes": {}
     }
   },
-  "total_flos": 1.1159446583771136e+19,
+  "total_flos": 1.1779415838425088e+19,
   "train_batch_size": 64,
   "trial_name": null,
   "trial_params": null