From 997fdd3770cf6546e1d6f087fd8e4c16f486b19a Mon Sep 17 00:00:00 2001
From: bishe <123456789@163.com>
Date: Fri, 7 Mar 2025 19:25:25 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E4=BA=86D=E8=BF=94=E5=9B=9Escores,=20?=
 =?UTF-8?q?features?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/networks.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/models/networks.py b/models/networks.py
index 7519e80..3c29522 100644
--- a/models/networks.py
+++ b/models/networks.py
@@ -1413,12 +1413,11 @@ class MLPDiscriminator(nn.Module):
         self.dropout = nn.Dropout(dropout)
 
     def forward(self, x):
-        x = self.linear1(x)
-        x = self.activation(x)
+        features = self.linear1(x)         # 中间特征，即 D_real 或 D_fake
+        x = self.activation(features)
         x = self.dropout(x)
-        x = self.linear2(x)
-        return self.dropout(x)
-
+        scores = self.linear2(x)           # 最终分数，即 real_scores 或 fake_scores
+        return scores, features
 
 class NLayerDiscriminator(nn.Module):
     """Defines a PatchGAN discriminator"""