diff --git a/stargan/model.py b/stargan/model.py
index 82f476d..79d2f73 100644
--- a/stargan/model.py
+++ b/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