diff --git a/stargan/solver.py b/stargan/solver.py index 65797d3..e60b7e5 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -621,7 +621,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)