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

Phase 5 fixing VAE implementation

Browse Source
This commit is contained in:
Johnny Fernandes
2026-05-03 22:21:21 +01:00
parent bda0745ecc
commit 801bb85258
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/configs/phase5/p5_vae.json
+1 -1
View File
@@ -12,7 +12,7 @@
"lr_d": 1e-4, "lr_d": 1e-4,
"beta_kl": 0.25, "beta_kl": 0.25,
"lambda_perceptual": 0.1, "lambda_perceptual": 0.1,
"lambda_adversarial": 0.1, "lambda_adversarial": 0.01,
"ndf_patch": 64, "ndf_patch": 64,
"sample_interval": 10, "sample_interval": 10,
"fid_interval": 25, "fid_interval": 25,
generator/src/training/trainer.py
+4
View File
@@ -1,5 +1,6 @@
import os import os
import time import time
import warnings
from pathlib import Path from pathlib import Path
import torch import torch
@@ -12,6 +13,9 @@ from tqdm import tqdm
from src.training.ema import EMA from src.training.ema import EMA
from src.training.fid import FIDEvaluator 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"): if hasattr(torch.amp, "GradScaler"):
_GradScaler = torch.amp.GradScaler _GradScaler = torch.amp.GradScaler
_autocast = torch.amp.autocast _autocast = torch.amp.autocast