From 80f6055cef8b6a9c6320e89a4eb3f7fa5626b085 Mon Sep 17 00:00:00 2001
From: Nataniel Ruiz <nruiz@Nataniels-MacBook-Pro.local>
Date: Thu, 9 Jan 2020 12:23:47 -0400
Subject: [PATCH] GANimation conditional attacks

---
 ganimation/attacks.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/ganimation/attacks.py b/ganimation/attacks.py
index d6918f4..a80fb0c 100644
--- a/ganimation/attacks.py
+++ b/ganimation/attacks.py
@@ -129,12 +129,11 @@ 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(out, y)
                 full_loss += loss
 
+            self.model.zero_grad()
             full_loss.backward()
             grad = X.grad