From 22fa209b26bd9fbb417f465c37babe0ad7e10bdf Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 26 Dec 2019 11:22:59 -0400 Subject: [PATCH] next --- stargan/solver.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/stargan/solver.py b/stargan/solver.py index 2edd809..775c543 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -609,20 +609,20 @@ class Solver(object): # Translate images. x_fake_list = [x_real] - x_advs = [] - if i == 0: - for idx, c_trg in enumerate(c_trg_list): - x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg) - x_advs.append((x_adv, perturb)) - break + # x_advs = [] + # if i == 0: + # for idx, c_trg in enumerate(c_trg_list): + # x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg) + # x_advs.append((x_adv, perturb)) + # break for idx, c_trg in enumerate(c_trg_list): 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 = x_real + perturb # x_adv = self.blur_tensor(x_adv) # Metrics