GANimation conditional attacks

2020-01-09 17:10:53 -04:00
parent 91ad218dc3
commit 79bd0e7874
2 changed files with 4 additions and 4 deletions
@@ -30,7 +30,7 @@ class LinfPGDAttack(object):
            out = imFromAttReg(output_att, output_img, X)

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