From e7fcfcf8b6c4474a2a8f72e090b73a1a59fb95c2 Mon Sep 17 00:00:00 2001
From: MSXYZ-GenAI <msahin@berksan.com>
Date: Wed, 22 Apr 2026 14:02:18 +0300
Subject: [PATCH] Add Trainer script

---
 Trainer/Trainer.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/Trainer/Trainer.py b/Trainer/Trainer.py
index 0e74eb8..a9c65ae 100644
--- a/Trainer/Trainer.py
+++ b/Trainer/Trainer.py
@@ -277,7 +277,7 @@ def train(
     lr             = 3e-4,
     patience       = 12,
     w_pixel        = 1.0,
-    w_edge         = 0.5,
+    w_edge         = 0.8,
     w_freq         = 0.1,
     w_luma         = 0.8,
     w_smooth       = 0.3,
@@ -351,6 +351,10 @@ def train(
         if avg < best_loss:
             best_loss  = avg
             best_state = {k: v.clone() for k, v in model.state_dict().items()}
+            
+            # Writes the file to disk from each new record
+            torch.save(best_state, output_path) 
+            print(f" [SYSTEM] Updated to the best model disk: {output_path} (Loss: {avg:.5f})")
 
         if stopper.step(avg):
             print(f"\n[DLAA] Early stop at epoch {epoch+1} (no improvement for {patience} epochs)")
@@ -394,7 +398,7 @@ if __name__ == "__main__":
             lr           = 3e-4,
             patience     = 12,
             w_pixel      = 1.0,
-            w_edge       = 0.5,
+            w_edge       = 0.8,
             w_freq       = 0.1,
             w_luma       = 0.8,
             w_smooth     = 0.3,