From 8233dde691f8e2f66f0785cdb085bb893081c593 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Mon, 30 Dec 2019 13:57:20 -0400 Subject: [PATCH] next --- stargan/solver.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/stargan/solver.py b/stargan/solver.py index e60b7e5..6458439 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -611,6 +611,7 @@ class Solver(object): # x_advs = [] if i == 0: + print(i) _, perturb = pgd_attack.perturb(x_real, black, c_trg_list[0]) # for idx, c_trg in enumerate(c_trg_list): # x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg) @@ -621,7 +622,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, black, c_trg) + # x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg) # _, perturb = x_advs[idx] x_adv = x_real + perturb # x_adv = self.blur_tensor(x_adv)