Merge branch 'main' of github.com:nnaakkaaii/D-3DVRVAE

nnaakkaaii · Jun 30, 2024 · f9274aa · f9274aa
2 parents d373023 + b829786
commit f9274aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hrdae/models/gan_model.py b/hrdae/models/gan_model.py
@@ -109,7 +109,7 @@ def train(
                 loss_g_basic = self.criterion(y, xp, latent=latent)
                 loss_g_adv = self.criterion_g(y_pred, torch.ones_like(y_pred))
 
-                loss_g = loss_g_basic + 0.01 * loss_g_adv
+                loss_g = loss_g_basic + 0.001 * loss_g_adv
                 loss_g.backward()
                 self.optimizer_g.step()