Phase 5 preparation

2026-05-03 18:00:29 +01:00
parent 8c0e845b5c
commit 57cf5261f2
1 changed files with 1 additions and 1 deletions
@@ -276,8 +276,8 @@ def train_wgan(
    ema = EMA(generator, decay=ema_decay)

    if hasattr(torch, "compile"):
+        # critic excluded — GP requires double backward which torch.compile doesn't support
        generator = torch.compile(generator)
-        critic    = torch.compile(critic)

    # Fixed noise for consistent sample tracking across epochs
    fixed_noise = torch.randn(16, latent_dim, 1, 1, device=device)