From 73ab898ee8a2aee70647905ac723df0877486165 Mon Sep 17 00:00:00 2001
From: MSXYZ-GenAI <msahin@berksan.com>
Date: Wed, 22 Apr 2026 13:28:57 +0300
Subject: [PATCH] Add Trainer script

---
 Trainer/Trainer.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/Trainer/Trainer.py b/Trainer/Trainer.py
index 38e1083..0e74eb8 100644
--- a/Trainer/Trainer.py
+++ b/Trainer/Trainer.py
@@ -276,13 +276,12 @@ def train(
     epochs         = 100,
     lr             = 3e-4,
     patience       = 12,
-    # loss weights
     w_pixel        = 1.0,
     w_edge         = 0.5,
     w_freq         = 0.1,
     w_luma         = 0.8,
     w_smooth       = 0.3,
-    w_perceptual   = 0.05,
+    w_perceptual   = 0.1,
 ):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     print(f"[DLAA] Device       : {device}")
@@ -339,7 +338,7 @@ def train(
         lr_now = scheduler.get_last_lr()[0]
         marker = " ★" if avg < best_loss else ""
 
-        # Detailed loss breakdown every 5 epochs
+        # Detailed loss every 5 epochs
         if (epoch + 1) % 5 == 0 or epoch == 0:
             breakdown = "  |  " + "  ".join(
                 f"{k}:{sub_totals[k]/n:.4f}" for k in sub_totals
@@ -399,7 +398,7 @@ if __name__ == "__main__":
             w_freq       = 0.1,
             w_luma       = 0.8,
             w_smooth     = 0.3,
-            w_perceptual = 0.05,
+            w_perceptual = 0.1,
         )
 
         print("\nTraining finished successfully.")