From 695facd227b95362b167983e7575202215784f35 Mon Sep 17 00:00:00 2001
From: the-database <25811902+the-database@users.noreply.github.com>
Date: Mon, 6 Jan 2025 21:41:00 -0500
Subject: [PATCH] fix for metagan discriminator

---
 traiNNer/models/sr_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/traiNNer/models/sr_model.py b/traiNNer/models/sr_model.py
index 9359be06..95d50381 100644
--- 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)