GANimation conditional attacks

2020-01-09 18:10:46 -04:00
parent 5ede7105ff
commit f1299dcd40
2 changed files with 15 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