GANimation conditional attacks

2020-01-09 12:22:22 -04:00
parent c824dccb6e
commit 77f174fc9b
1 changed files with 2 additions and 2 deletions
@@ -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)