update loss_g_adv ratio

nnaakkaaii · Jun 30, 2024 · b829786 · b829786
1 parent acac082
commit b829786
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()