GANimation conditional attacks

2020-01-09 12:27:28 -04:00
parent f02039f6eb
commit a4ce503238
1 changed files with 2 additions and 2 deletions
@@ -119,9 +119,9 @@ class LinfPGDAttack(object):
        X = X_nat.clone().detach_()

        J = c_trg.size(0)
-        full_loss = 0.0
-
+        
        for i in range(self.k):
+            full_loss = 0.0
            X.requires_grad = True
            self.model.zero_grad()