fix for metagan discriminator

the-database · Jan 7, 2025 · 695facd · 695facd
1 parent 72e75e6
commit 695facd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/traiNNer/models/sr_model.py b/traiNNer/models/sr_model.py
@@ -357,8 +357,10 @@ def optimize_parameters(
             for label, loss in self.losses.items():
                 if label == "l_g_gan":
                     assert self.net_d is not None
+                    self.net_d.eval()
                     fake_g_pred = self.net_d(self.output)
                     l_g_loss = loss(fake_g_pred, True, is_disc=False)
+                    self.net_d.train()
                 else:
                     l_g_loss = loss(self.output, self.gt)