next

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