From f659e2af195ed9145d1160ecba1c0207925fc817 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 9 Jan 2020 14:13:17 -0400 Subject: [PATCH] GANimation conditional attacks --- ganimation/attacks.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/ganimation/attacks.py b/ganimation/attacks.py index b2c0ea5..798dde1 100644 --- a/ganimation/attacks.py +++ b/ganimation/attacks.py @@ -30,8 +30,8 @@ class LinfPGDAttack(object): out = imFromAttReg(output_att, output_img, X) self.model.zero_grad() - # loss = -self.loss_fn(output_att, y) + self.loss_fn(output_img, 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 @@ -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 @@ -131,8 +131,8 @@ class LinfPGDAttack(object): out = imFromAttReg(output_att, output_img, X) - # 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) full_loss += loss full_loss.backward()