GANimation conditional attacks

2020-01-09 12:15:56 -04:00
parent b0c3723d11
commit 384422955a
1 changed files with 4 additions and 3 deletions
@@ -96,8 +96,8 @@ class LinfPGDAttack(object):

            self.model.zero_grad()

-            # Away from black
-            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

@@ -131,7 +131,8 @@ class LinfPGDAttack(object):

                self.model.zero_grad()

-                loss = -self.loss_fn(output_att, y)
+                # loss = -self.loss_fn(output_att, y)
+                loss = -self.loss_fn(out, y)
                full_loss += loss

            full_loss.backward()