From 7b496aab8563cd8ae76e6fa0bcadc0f5877b148f Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Tue, 24 Dec 2019 18:59:55 -0400 Subject: [PATCH] next --- stargan/solver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stargan/solver.py b/stargan/solver.py index 10e8d07..67b0101 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -613,7 +613,7 @@ class Solver(object): with torch.no_grad(): gen_noattack, gen_noattack_feats = self.G(x_real, c_trg) # Attack - x_adv, perturb = pgd_attack.perturb(x_real, gen_noattack, c_trg) + x_adv, perturb = pgd_attack.perturb(x_real, x_real, c_trg) # x_adv = x_real + perturb # x_adv = self.blur_tensor(x_adv)