jalf/DRL_PROJ
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Phase 5 preparation

Browse Source
This commit is contained in:
Johnny Fernandes
2026-05-03 18:00:29 +01:00
parent 8c0e845b5c
commit 57cf5261f2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/src/training/trainer.py
+1 -1
View File
@@ -276,8 +276,8 @@ def train_wgan(
ema = EMA(generator, decay=ema_decay) ema = EMA(generator, decay=ema_decay)
if hasattr(torch, "compile"): if hasattr(torch, "compile"):
# critic excluded — GP requires double backward which torch.compile doesn't support
generator = torch.compile(generator) generator = torch.compile(generator)
critic = torch.compile(critic)
# Fixed noise for consistent sample tracking across epochs # Fixed noise for consistent sample tracking across epochs
fixed_noise = torch.randn(16, latent_dim, 1, 1, device=device) fixed_noise = torch.randn(16, latent_dim, 1, 1, device=device)