bugfix

nocotan · Dec 29, 2020 · 3a30234 · 3a30234
1 parent 3870296
commit 3a30234
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/wgan_gp.py b/models/wgan_gp.py
@@ -186,8 +186,8 @@ def configure_optimizers(self):
         opt_g = torch.optim.Adam(self.generator.parameters(), lr=lr, betas=(b1, b2))
         opt_d = torch.optim.Adam(self.discriminator.parameters(), lr=lr, betas=(b1, b2))
         return (
-            {'optimizer': opt_g, 'frequency': n_critic},
-            {'optimizer': opt_d, 'frequency': 1}
+            {'optimizer': opt_g, 'frequency': 1},
+            {'optimizer': opt_d, 'frequency': n_critic}
         )
 
     def train_dataloader(self):