GANimation conditional attacks

2020-01-09 12:23:47 -04:00
parent 77f174fc9b
commit 80f6055cef
1 changed files with 1 additions and 2 deletions
@@ -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