End of training

README.md

@@ -1,4 +1,6 @@
 ---
+language:
+- id
 license: mit
 base_model: indolem/indobert-base-uncased
 tags:

all_results.json

@@ -1,49 +1,49 @@
 {
-    "epoch": 100.0,
-    "eval_LOCATION_f1": 0.9491525423728815,
+    "epoch": 5.0,
+    "eval_LOCATION_f1": 0.9527896995708154,
     "eval_LOCATION_number": 116,
-    "eval_LOCATION_precision": 0.9333333333333333,
-    "eval_LOCATION_recall": 0.9655172413793104,
-    "eval_ORGANIZATION_f1": 0.9487179487179488,
+    "eval_LOCATION_precision": 0.9487179487179487,
+    "eval_LOCATION_recall": 0.9568965517241379,
+    "eval_ORGANIZATION_f1": 0.9517684887459807,
     "eval_ORGANIZATION_number": 158,
-    "eval_ORGANIZATION_precision": 0.961038961038961,
+    "eval_ORGANIZATION_precision": 0.9673202614379085,
     "eval_ORGANIZATION_recall": 0.9367088607594937,
     "eval_PERSON_f1": 0.9840000000000001,
     "eval_PERSON_number": 124,
     "eval_PERSON_precision": 0.9761904761904762,
     "eval_PERSON_recall": 0.9919354838709677,
-    "eval_loss": 0.08658473193645477,
+    "eval_loss": 0.039770063012838364,
     "eval_overall_accuracy": 0.9912160307438924,
-    "eval_overall_f1": 0.9598997493734336,
-    "eval_overall_precision": 0.9575,
-    "eval_overall_recall": 0.9623115577889447,
-    "eval_runtime": 0.6335,
+    "eval_overall_f1": 0.9622166246851386,
+    "eval_overall_precision": 0.9646464646464646,
+    "eval_overall_recall": 0.9597989949748744,
+    "eval_runtime": 1.1963,
     "eval_samples": 170,
-    "eval_samples_per_second": 268.358,
-    "eval_steps_per_second": 4.736,
-    "predict_LOCATION_f1": 0.9915966386554622,
+    "eval_samples_per_second": 142.108,
+    "eval_steps_per_second": 2.508,
+    "predict_LOCATION_f1": 0.9771309771309771,
     "predict_LOCATION_number": 237,
-    "predict_LOCATION_precision": 0.9874476987447699,
-    "predict_LOCATION_recall": 0.9957805907172996,
-    "predict_ORGANIZATION_f1": 0.9906759906759907,
+    "predict_LOCATION_precision": 0.9631147540983607,
+    "predict_LOCATION_recall": 0.9915611814345991,
+    "predict_ORGANIZATION_f1": 0.9882903981264638,
     "predict_ORGANIZATION_number": 431,
-    "predict_ORGANIZATION_precision": 0.9953161592505855,
-    "predict_ORGANIZATION_recall": 0.9860788863109049,
-    "predict_PERSON_f1": 0.9986541049798116,
+    "predict_ORGANIZATION_precision": 0.9976359338061466,
+    "predict_ORGANIZATION_recall": 0.9791183294663574,
+    "predict_PERSON_f1": 0.9919354838709676,
     "predict_PERSON_number": 371,
-    "predict_PERSON_precision": 0.9973118279569892,
-    "predict_PERSON_recall": 1.0,
-    "predict_loss": 0.009614275768399239,
-    "predict_overall_accuracy": 0.9989429175475687,
-    "predict_overall_f1": 0.9937409725565719,
-    "predict_overall_precision": 0.9942196531791907,
-    "predict_overall_recall": 0.9932627526467758,
-    "predict_runtime": 1.734,
-    "predict_samples_per_second": 245.097,
-    "predict_steps_per_second": 4.037,
-    "train_loss": 0.005912137792135278,
-    "train_runtime": 2272.4202,
+    "predict_PERSON_precision": 0.9892761394101877,
+    "predict_PERSON_recall": 0.9946091644204852,
+    "predict_loss": 0.010024257004261017,
+    "predict_overall_accuracy": 0.9975687103594081,
+    "predict_overall_f1": 0.987012987012987,
+    "predict_overall_precision": 0.9865384615384616,
+    "predict_overall_recall": 0.987487969201155,
+    "predict_runtime": 3.6213,
+    "predict_samples_per_second": 117.362,
+    "predict_steps_per_second": 1.933,
+    "train_loss": 0.07880291553835074,
+    "train_runtime": 223.1323,
     "train_samples": 1531,
-    "train_samples_per_second": 67.373,
-    "train_steps_per_second": 4.225
+    "train_samples_per_second": 34.307,
+    "train_steps_per_second": 2.151
 }

eval_results.json

@@ -1,24 +1,24 @@
 {
-    "epoch": 100.0,
-    "eval_LOCATION_f1": 0.9491525423728815,
+    "epoch": 5.0,
+    "eval_LOCATION_f1": 0.9527896995708154,
     "eval_LOCATION_number": 116,
-    "eval_LOCATION_precision": 0.9333333333333333,
-    "eval_LOCATION_recall": 0.9655172413793104,
-    "eval_ORGANIZATION_f1": 0.9487179487179488,
+    "eval_LOCATION_precision": 0.9487179487179487,
+    "eval_LOCATION_recall": 0.9568965517241379,
+    "eval_ORGANIZATION_f1": 0.9517684887459807,
     "eval_ORGANIZATION_number": 158,
-    "eval_ORGANIZATION_precision": 0.961038961038961,
+    "eval_ORGANIZATION_precision": 0.9673202614379085,
     "eval_ORGANIZATION_recall": 0.9367088607594937,
     "eval_PERSON_f1": 0.9840000000000001,
     "eval_PERSON_number": 124,
     "eval_PERSON_precision": 0.9761904761904762,
     "eval_PERSON_recall": 0.9919354838709677,
-    "eval_loss": 0.08658473193645477,
+    "eval_loss": 0.039770063012838364,
     "eval_overall_accuracy": 0.9912160307438924,
-    "eval_overall_f1": 0.9598997493734336,
-    "eval_overall_precision": 0.9575,
-    "eval_overall_recall": 0.9623115577889447,
-    "eval_runtime": 0.6335,
+    "eval_overall_f1": 0.9622166246851386,
+    "eval_overall_precision": 0.9646464646464646,
+    "eval_overall_recall": 0.9597989949748744,
+    "eval_runtime": 1.1963,
     "eval_samples": 170,
-    "eval_samples_per_second": 268.358,
-    "eval_steps_per_second": 4.736
+    "eval_samples_per_second": 142.108,
+    "eval_steps_per_second": 2.508
 }

predict_results.json

@@ -1,22 +1,22 @@
 {
-    "predict_LOCATION_f1": 0.9915966386554622,
+    "predict_LOCATION_f1": 0.9771309771309771,
     "predict_LOCATION_number": 237,
-    "predict_LOCATION_precision": 0.9874476987447699,
-    "predict_LOCATION_recall": 0.9957805907172996,
-    "predict_ORGANIZATION_f1": 0.9906759906759907,
+    "predict_LOCATION_precision": 0.9631147540983607,
+    "predict_LOCATION_recall": 0.9915611814345991,
+    "predict_ORGANIZATION_f1": 0.9882903981264638,
     "predict_ORGANIZATION_number": 431,
-    "predict_ORGANIZATION_precision": 0.9953161592505855,
-    "predict_ORGANIZATION_recall": 0.9860788863109049,
-    "predict_PERSON_f1": 0.9986541049798116,
+    "predict_ORGANIZATION_precision": 0.9976359338061466,
+    "predict_ORGANIZATION_recall": 0.9791183294663574,
+    "predict_PERSON_f1": 0.9919354838709676,
     "predict_PERSON_number": 371,
-    "predict_PERSON_precision": 0.9973118279569892,
-    "predict_PERSON_recall": 1.0,
-    "predict_loss": 0.009614275768399239,
-    "predict_overall_accuracy": 0.9989429175475687,
-    "predict_overall_f1": 0.9937409725565719,
-    "predict_overall_precision": 0.9942196531791907,
-    "predict_overall_recall": 0.9932627526467758,
-    "predict_runtime": 1.734,
-    "predict_samples_per_second": 245.097,
-    "predict_steps_per_second": 4.037
+    "predict_PERSON_precision": 0.9892761394101877,
+    "predict_PERSON_recall": 0.9946091644204852,
+    "predict_loss": 0.010024257004261017,
+    "predict_overall_accuracy": 0.9975687103594081,
+    "predict_overall_f1": 0.987012987012987,
+    "predict_overall_precision": 0.9865384615384616,
+    "predict_overall_recall": 0.987487969201155,
+    "predict_runtime": 3.6213,
+    "predict_samples_per_second": 117.362,
+    "predict_steps_per_second": 1.933
 }

predictions.txt

@@ -5,14 +5,14 @@ O B-LOCATION O O B-LOCATION O O O O O O O O
 O O O O O O O O O O O O O O O O
 O B-ORGANIZATION O O O O O O O B-PERSON O O O O O O
 O O O O B-PERSON O O O O O O O O O O O O O O O O O O B-PERSON O O O O O O O O
-B-PERSON O O O O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O O O O O B-ORGANIZATION O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O
+B-PERSON O O O O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O O O O O O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O
 O O B-LOCATION I-LOCATION O O O O O O O O O O B-LOCATION I-LOCATION O O O O O O O O
 B-LOCATION O O O O O O O B-LOCATION O O O
 O O O O O O O O O B-PERSON I-PERSON O O B-PERSON I-PERSON O
 O O O O O B-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O O O B-LOCATION I-LOCATION O
 B-PERSON O O O B-ORGANIZATION O O O O O O O O O B-LOCATION O O O O O O B-LOCATION O
 B-ORGANIZATION O O O O O O O O O O O O O O O O O O O O O
-O O O O O O O O O O B-LOCATION O O B-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O O O O O O
+O O O O O O O O O B-LOCATION I-LOCATION O O B-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O O O O O O
 O O O O O O O O O O O O B-PERSON O O O O O O B-LOCATION I-LOCATION I-LOCATION I-LOCATION O B-LOCATION O O O O O O
 O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O B-PERSON O
 O O B-ORGANIZATION O O O O O O O O
@@ -37,7 +37,7 @@ O O B-LOCATION O B-PERSON O O O O O O O O
 O B-ORGANIZATION O O O O O O O O O O O O O O
 O O O B-LOCATION O O O B-ORGANIZATION I-ORGANIZATION O O O O O O O O O O O O B-LOCATION O
 B-PERSON O O O O O O O O O O O O O
-O O O O O B-LOCATION O O O O O B-LOCATION O O O O O O O O O B-LOCATION O O B-LOCATION O B-LOCATION O B-LOCATION O B-LOCATION O B-PERSON O O O O O B-ORGANIZATION O O O O O O
+O O O O O B-LOCATION O O O O O B-LOCATION O O O O O O O O O B-LOCATION O O B-LOCATION O B-LOCATION O B-LOCATION O B-LOCATION O B-PERSON O O O O O B-ORGANIZATION O B-ORGANIZATION O O O O
 O O O O O O O O O O O O O B-ORGANIZATION O O O O O
 B-LOCATION O B-ORGANIZATION I-ORGANIZATION O O B-ORGANIZATION I-ORGANIZATION O O O O O O
 O O O O O O O O O O O O O O O O O O O B-PERSON O O O O O
@@ -78,7 +78,7 @@ B-ORGANIZATION O O O O O O O O O O
 O O O O B-PERSON O O O O O O O O O O O
 B-PERSON O O O O O O O O O O
 O O O O O O O O O B-ORGANIZATION I-ORGANIZATION B-ORGANIZATION I-ORGANIZATION O
-O O O O O O O O O B-PERSON O B-ORGANIZATION O B-ORGANIZATION I-ORGANIZATION O O O O O O
+O O O O O O O O O B-PERSON O B-LOCATION O B-LOCATION I-LOCATION O O O O O O
 O O O O B-LOCATION I-LOCATION I-LOCATION I-LOCATION O O O O O O O O
 O B-ORGANIZATION I-ORGANIZATION O B-PERSON I-PERSON O O O O B-PERSON I-PERSON O B-PERSON O
 O O O O O O O O O O O O O O O O O O B-PERSON O
@@ -151,7 +151,7 @@ O O B-ORGANIZATION O B-ORGANIZATION O O O O O O O O O O O O O O O O O
 O O O O O O O O O O O O O O O O O O O O B-PERSON O
 O B-ORGANIZATION O O O O O O O O O O O
 O O O O O O O O O O O O O O O O O O O B-PERSON O O O O O O O O B-LOCATION O
-O O B-ORGANIZATION O O B-PERSON I-PERSON I-PERSON I-PERSON O O O B-LOCATION O
+O O B-PERSON O O B-PERSON I-PERSON I-PERSON I-PERSON O O O B-LOCATION O
 O O O O O O O O O B-LOCATION O O O O O O O
 O O O O O B-PERSON O B-PERSON O O O O O O O O O O
 O O O O O O O O O O O O O O O O O O O B-ORGANIZATION I-ORGANIZATION O O O O O O O O O O O O O O O O O O
@@ -168,7 +168,7 @@ B-PERSON O O O O O O O O B-PERSON I-PERSON O O O
 O O B-LOCATION B-PERSON I-PERSON I-PERSON O B-PERSON O O O O O O O O O B-LOCATION O O O O O O O O O O O
 O O B-PERSON I-PERSON O O O O O O O O O O O O O O
 B-PERSON I-PERSON I-PERSON O O O O O O O O O B-PERSON I-PERSON O O O O O O B-LOCATION O O
-O O O O O B-ORGANIZATION B-ORGANIZATION I-ORGANIZATION O O O O
+O O O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O O O O
 O O O B-PERSON O O B-LOCATION O O O O O O O B-LOCATION I-LOCATION O O O O O
 O O O O O O O O O O B-PERSON I-PERSON O B-PERSON I-PERSON O O O O O O O O O B-LOCATION O O O O O O O O
 O O O B-LOCATION I-LOCATION O O B-LOCATION B-PERSON I-PERSON O O O O O B-ORGANIZATION O O B-LOCATION I-LOCATION O
@@ -212,7 +212,7 @@ B-LOCATION I-LOCATION O O O O O O O O O O O
 O O O B-ORGANIZATION O O O B-PERSON I-PERSON O O O O O O O O B-PERSON I-PERSON O
 O O O O O O O O B-PERSON O B-PERSON O O O O O O O O B-PERSON I-PERSON O B-PERSON I-PERSON O O O B-ORGANIZATION O
 O O O B-PERSON I-PERSON O B-ORGANIZATION I-ORGANIZATION O O O O B-ORGANIZATION O
-O O B-LOCATION I-LOCATION O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O
+O O B-LOCATION I-LOCATION O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O I-ORGANIZATION O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O
 O O O O O B-ORGANIZATION O O O O O O O O B-ORGANIZATION B-PERSON I-PERSON O O O O O O O O O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O O O B-LOCATION I-LOCATION
 B-PERSON O O O O O O O O O O O O O O O O B-PERSON O O O O
 B-ORGANIZATION I-ORGANIZATION O O O O O O O O O O O O
@@ -287,7 +287,7 @@ O O O B-LOCATION O O O B-LOCATION O O O O O O O O B-LOCATION O O O O O B-LOCATIO
 O B-PERSON O O O O O O O O O O O O O O O O O O O B-PERSON O B-PERSON O O O B-ORGANIZATION O B-PERSON I-PERSON O
 O B-PERSON I-PERSON O O O O O B-PERSON I-PERSON O O O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O O O
 B-ORGANIZATION O O O O O O O O O O O O O O O
-O O O O O O O O B-PERSON I-PERSON I-PERSON O O O O O O O O O B-PERSON O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O O O O O O O O O O O O O O O O O O O B-PERSON I-PERSON I-PERSON O
+O O O O O O O O B-PERSON I-PERSON I-PERSON O O O O O O O O O B-LOCATION O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O O O O O O O O O O O O O O O O O O O B-PERSON I-PERSON I-PERSON O
 O B-ORGANIZATION O O O O O O O O O O O O O
 O O O B-ORGANIZATION O O O O O O B-PERSON O B-LOCATION O O O O O O
 O O O O O O O O O O O O O B-LOCATION O O O O O
@@ -299,7 +299,7 @@ O O O O O B-ORGANIZATION O O O O O O O O O O O O O O O O O O O O B-ORGANIZATION
 O O B-LOCATION I-LOCATION I-LOCATION O O O O O O O O O O O O B-PERSON O
 O O O O O O O O O O O O O O O O O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O
 O O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O O O O O O O O O O O O B-ORGANIZATION O
-O O O O O B-ORGANIZATION B-PERSON I-PERSON O O O O B-ORGANIZATION O O O B-LOCATION O O O O O O
+O O O O O B-ORGANIZATION B-PERSON I-PERSON O O O O B-ORGANIZATION O B-LOCATION O B-LOCATION O O O O O O
 B-LOCATION O O O B-ORGANIZATION O B-PERSON I-PERSON O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-PERSON I-PERSON O
 O O O O O O O O O O O O O O O O O O O O O O O O O O B-PERSON O
 O O O O B-ORGANIZATION I-ORGANIZATION O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O O O O O O O O
@@ -309,7 +309,7 @@ O O O O O O O O O O O O O O O O B-LOCATION I-LOCATION O B-LOCATION I-LOCATION O
 B-PERSON O O O O O B-ORGANIZATION O O O O O O O O O O O O O O O O O O O
 O O O O O B-PERSON O O B-ORGANIZATION O O O O O O O O O
 O O O O O O O O O O O O O O O O O B-PERSON I-PERSON O
-O O B-LOCATION B-ORGANIZATION I-ORGANIZATION O O O O O O O O O O O O O
+O O B-LOCATION B-PERSON I-PERSON O O O O O O O O O O O O O
 O O O O O O O O O B-PERSON O O O O O
 O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O B-PERSON I-PERSON O O O O O O O O O O O O O O O
 O O O O O O O O O O O O O B-LOCATION O O O O O B-PERSON O B-PERSON O B-LOCATION I-LOCATION O B-LOCATION I-LOCATION I-LOCATION I-LOCATION O B-LOCATION I-LOCATION O O O O O O
@@ -325,7 +325,7 @@ O O O O O O O O O O O O B-PERSON O O O O O O O O O B-PERSON O
 O O O B-PERSON I-PERSON O O O O B-PERSON I-PERSON O O O B-PERSON I-PERSON O O O O O O O O B-PERSON O O O O O O O O B-PERSON O
 B-ORGANIZATION O O O O O O O O O O O O O O O O O
 O O B-ORGANIZATION O O O O O O O O O O O O B-ORGANIZATION O O O B-ORGANIZATION O O O O O O O O O O O
-O O O O B-PERSON O B-PERSON O O O O O O O O O O O O O O B-PERSON I-PERSON O B-PERSON I-PERSON O
+O O O O B-PERSON O B-PERSON O O O O O B-LOCATION O O O O O O O O B-PERSON I-PERSON O B-PERSON I-PERSON O
 O B-PERSON I-PERSON O O O O O B-PERSON I-PERSON O O O O O O O O B-LOCATION I-LOCATION O B-LOCATION I-LOCATION O B-LOCATION O
 O O O O O B-ORGANIZATION B-LOCATION I-LOCATION O B-PERSON I-PERSON O
 B-LOCATION O O O O O O O O O O O
@@ -367,7 +367,7 @@ O B-LOCATION I-LOCATION O O O O O O O O O O O O O O O O O O O O O O O
 O O O O O O O B-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O B-ORGANIZATION O B-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O B-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION I-ORGANIZATION O
 O O O O O O O B-ORGANIZATION I-ORGANIZATION B-PERSON O O B-PERSON O O O O O O
 O O O O O O O O O O O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O B-ORGANIZATION O O O O B-LOCATION I-LOCATION O
-O O O O O O O O O O O O O O O O O O O O B-PERSON I-PERSON O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O O O O B-ORGANIZATION O
+O O O O O O O O O O O O O O O O O O O O O B-PERSON O B-ORGANIZATION I-ORGANIZATION I-ORGANIZATION I-ORGANIZATION O O O O B-ORGANIZATION O
 O O O O O O O O O O O O O O O O O O O O O O O O O O O O B-PERSON O
 O O O O O O O B-ORGANIZATION O O O B-ORGANIZATION O
 B-PERSON O O O O O O O O B-ORGANIZATION O O B-LOCATION I-LOCATION O B-LOCATION I-LOCATION I-LOCATION O B-LOCATION O B-LOCATION I-LOCATION O O O O O O

runs/May28_20-05-39_indolem-petl-vm/events.out.tfevents.1716927018.indolem-petl-vm.213003.1

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

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 100.0,
-    "train_loss": 0.005912137792135278,
-    "train_runtime": 2272.4202,
+    "epoch": 5.0,
+    "train_loss": 0.07880291553835074,
+    "train_runtime": 223.1323,
     "train_samples": 1531,
-    "train_samples_per_second": 67.373,
-    "train_steps_per_second": 4.225
+    "train_samples_per_second": 34.307,
+    "train_steps_per_second": 2.151
 }