diff --git a/ganimation/attacks.py b/ganimation/attacks.py
index c879e6a..d6918f4 100644
--- a/ganimation/attacks.py
+++ b/ganimation/attacks.py
@@ -122,9 +122,9 @@ class LinfPGDAttack(object):
         full_loss = 0.0
 
         for i in range(self.k):
+            X.requires_grad = True
             for j in range(J):
-                # print(i)
-                X.requires_grad = True
+                print(i, j)
                 output_att, output_img = self.model(X, c_trg[j,:].unsqueeze(0))
 
                 out = imFromAttReg(output_att, output_img, X)