fix bug

components/Generator.py

@@ -5,7 +5,7 @@
 # Created Date: Sunday January 16th 2022
 # Author: Chen Xuanhong
 # Email: chenxuanhongzju@outlook.com
-# Last Modified:  Sunday, 16th January 2022 11:42:14 pm
+# Last Modified:  Sunday, 13th February 2022 2:03:21 am
 # Modified By: Chen Xuanhong
 # Copyright (c) 2022 Shanghai Jiao Tong University
 #############################################################
@@ -166,21 +166,21 @@ class Generator(nn.Module):
     #         if isinstance(layer,nn.Conv2d):
     #             nn.init.xavier_uniform_(layer.weight)
 
-    def forward(self, input, id):
-        x = input  # 3*224*224
-        skip1 = self.first_layer(x)
-        skip2 = self.down1(skip1)
-        skip3 = self.down2(skip2)
-        skip4 = self.down3(skip3)
-        res   = self.down4(skip4)
+    def forward(self, img, id):
+        # x = input  # 3*224*224
+        res = self.first_layer(img)
+        res = self.down1(res)
+        res = self.down2(res)
+        res = self.down3(res)
+        res = self.down4(res)
 
         for i in range(len(self.BottleNeck)):
-            x = self.BottleNeck[i](res, id)
+            res = self.BottleNeck[i](res, id)
 
-        x = self.up4(x)
-        x = self.up3(x)
-        x = self.up2(x)
-        x = self.up1(x)
-        x = self.last_layer(x)
+        res = self.up4(res)
+        res = self.up3(res)
+        res = self.up2(res)
+        res = self.up1(res)
+        res = self.last_layer(res)
 
-        return x
+        return res