From f02039f6eb23358f7a9b62354175f9ad715d798f Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 9 Jan 2020 12:26:34 -0400 Subject: [PATCH] GANimation conditional attacks --- ganimation/attacks.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ganimation/attacks.py b/ganimation/attacks.py index a80fb0c..e43bba0 100644 --- a/ganimation/attacks.py +++ b/ganimation/attacks.py @@ -123,6 +123,8 @@ class LinfPGDAttack(object): for i in range(self.k): X.requires_grad = True + self.model.zero_grad() + for j in range(J): print(i, j) output_att, output_img = self.model(X, c_trg[j,:].unsqueeze(0)) @@ -133,7 +135,6 @@ class LinfPGDAttack(object): loss = -self.loss_fn(out, y) full_loss += loss - self.model.zero_grad() full_loss.backward() grad = X.grad