next

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)

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