Phase 5 fixing VAE implementation

generator/src/training/trainer.py

@@ -1,5 +1,6 @@
 import os
 import time
+import warnings
 from pathlib import Path
 
 import torch
@@ -12,6 +13,9 @@ from tqdm import tqdm
 from src.training.ema import EMA
 from src.training.fid import FIDEvaluator
 
+# LambdaLR calls step() once in __init__ before any optimizer step — false positive
+warnings.filterwarnings("ignore", message="Detected call of `lr_scheduler.step\\(\\)` before `optimizer.step\\(\\)`")
+
 if hasattr(torch.amp, "GradScaler"):
     _GradScaler = torch.amp.GradScaler
     _autocast = torch.amp.autocast