next

stargan/model.py

@@ -66,8 +66,10 @@ class Generator(nn.Module):
 
         # Get intermediate feature maps
         for layer in self.main:
+            print('test')
             x = layer(x)
             feature_maps.append(x)
+        print(feature_maps.shape)
 
         return x, feature_maps