From 1b162241ed8aa14b75f8428579bd58099f052a26 Mon Sep 17 00:00:00 2001
From: Nataniel Ruiz <nruiz@Nataniels-MacBook-Pro.local>
Date: Mon, 23 Dec 2019 16:47:52 -0400
Subject: [PATCH] next

---
 stargan/attacks.py | 1 +
 stargan/model.py   | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

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