diff --git a/stargan/attacks.py b/stargan/attacks.py
index 7edf3bc..80424c7 100644
--- a/stargan/attacks.py
+++ b/stargan/attacks.py
@@ -29,6 +29,7 @@ class LinfPGDAttack(object):
             output, feats = self.model(X, c_trg)
 
             if self.feat:
+                print('self.feat ', self.feat)
                 output = feats[self.feat]
                 y = np.zeros(output.shape)
                 y = torch.FloatTensor(y).to(self.device)
diff --git a/stargan/model.py b/stargan/model.py
index 79d2f73..beb7d82 100644
--- a/stargan/model.py
+++ b/stargan/model.py
@@ -69,7 +69,7 @@ class Generator(nn.Module):
             print('test')
             x = layer(x)
             feature_maps.append(x)
-        print(feature_maps.shape)
+        print(len(feature_maps))
 
         return x, feature_maps