@@ -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)