Updating readme.

jzonthemtn · Aug 31, 2022 · a6d766f · a6d766f
1 parent 44947be
commit a6d766f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -17,7 +17,8 @@ python3 -m pip install transformers onnxruntime torch sklearn
 
 ## Evaluation
 
-| Version      | Evaluation |
+| Epochs      | 1 |
 | ----------- | ----------- |
 | 1      | {'eval_loss': 0.21822933852672577, 'eval_accuracy': 0.93224, 'eval_f1': 0.9321042084168337, 'eval_runtime': 225.0407, 'eval_samples_per_second': 111.091, 'eval_steps_per_second': 13.886}      |
+| 5 | {'eval_loss': 0.3719911277294159, 'eval_accuracy': 0.93136, 'eval_f1': 0.9316987740805606, 'eval_runtime': 235.7789, 'eval_samples_per_second': 106.032, 'eval_steps_per_second': 6.629, 'epoch': 5.0} |