Model save

all_results.json

@@ -1,4 +1,8 @@
 {
+    "epoch": 1.0,
     "eval_accuracy": 0.9532908704883227,
-    "eval_loss": 0.16175414621829987
+    "eval_loss": 0.16175416111946106,
+    "eval_runtime": 971.1712,
+    "eval_samples_per_second": 0.485,
+    "eval_steps_per_second": 0.061
 }

test_results.json

@@ -1,4 +1,8 @@
 {
+    "epoch": 1.0,
     "eval_accuracy": 0.9532908704883227,
-    "eval_loss": 0.16175414621829987
+    "eval_loss": 0.16175416111946106,
+    "eval_runtime": 971.1712,
+    "eval_samples_per_second": 0.485,
+    "eval_steps_per_second": 0.061
 }

trainer_state.json

@@ -1,6 +1,6 @@
 {
-  "best_metric": 0.9624664879356568,
-  "best_model_checkpoint": "videomae-base-finetuned-RealLifeViolenceSituations-subset/checkpoint-400",
+  "best_metric": 0.9625668449197861,
+  "best_model_checkpoint": "videomae-base-finetuned-RealLifeViolenceSituations-subset/checkpoint-801",
   "epoch": 1.00125,
   "global_step": 801,
   "is_hyper_param_search": false,
@@ -525,35 +525,44 @@
     },
     {
       "epoch": 1.0,
-      "eval_accuracy": 0.9532908704883227,
-      "eval_loss": 0.16175414621829987,
-      "eval_runtime": 1014.572,
-      "eval_samples_per_second": 0.464,
-      "eval_steps_per_second": 0.058,
+      "eval_accuracy": 0.9625668449197861,
+      "eval_loss": 0.14114385843276978,
+      "eval_runtime": 846.6552,
+      "eval_samples_per_second": 0.442,
+      "eval_steps_per_second": 0.056,
       "step": 801
     },
     {
       "epoch": 1.0,
-      "eval_accuracy": 0.9567779960707269,
-      "eval_loss": 0.14998553693294525,
-      "eval_runtime": 1046.019,
-      "eval_samples_per_second": 0.487,
-      "eval_steps_per_second": 0.061,
+      "step": 801,
+      "total_flos": 7.99475040605071e+18,
+      "train_loss": 2.0883763476406293e-06,
+      "train_runtime": 900.3937,
+      "train_samples_per_second": 7.108,
+      "train_steps_per_second": 0.889
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9585687382297552,
+      "eval_loss": 0.14402535557746887,
+      "eval_runtime": 1042.5133,
+      "eval_samples_per_second": 0.509,
+      "eval_steps_per_second": 0.064,
       "step": 801
     },
     {
       "epoch": 1.0,
       "eval_accuracy": 0.9532908704883227,
-      "eval_loss": 0.16175414621829987,
-      "eval_runtime": 943.6241,
-      "eval_samples_per_second": 0.499,
-      "eval_steps_per_second": 0.063,
+      "eval_loss": 0.16175416111946106,
+      "eval_runtime": 971.1712,
+      "eval_samples_per_second": 0.485,
+      "eval_steps_per_second": 0.061,
       "step": 801
     }
   ],
   "max_steps": 800,
   "num_train_epochs": 9223372036854775807,
-  "total_flos": 7.974813372619162e+18,
+  "total_flos": 7.99475040605071e+18,
   "trial_name": null,
   "trial_params": null
 }