Added epochs

2025-10-17 17:06:42 +02:00
parent 86a063591e
commit 540b78204c
2 changed files with 5 additions and 5 deletions
--- a/Project_Model/Libs/Training/loss_saver.py
+++ b/Project_Model/Libs/Training/loss_saver.py
@@ -3,8 +3,8 @@ import os
 class Log:
    def __init__(self, path):
        self.path = path
-        header = ["avg_txt","avg_enc","avg_dec","txt_loss","masking_loss","prediction_loss"]
-        
+        header = ["epoch","avg_txt","avg_enc","avg_dec","txt_loss","masking_loss","prediction_loss"]
+
        with open(self.path, "w", encoding="utf-8", newline="") as f:
            f.write(",".join(header) + "\n")