GANimation conditional attacks

2020-01-09 18:06:27 -04:00
parent bbaa697c36
commit 5ede7105ff
2 changed files with 16 additions and 15 deletions
@@ -96,8 +96,8 @@ class LinfPGDAttack(object):

            self.model.zero_grad()

-            loss = self.loss_fn(output_att, y)
-            # loss = -self.loss_fn(out, y)
+            # loss = self.loss_fn(output_att, y)
+            loss = -self.loss_fn(out, y)
            loss.backward()
            grad = X.grad