diff --git a/stargan/solver.py b/stargan/solver.py index d3e787c..b286035 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -615,8 +615,10 @@ class Solver(object): x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg) x_advs.append((x_adv, perturb)) break - + + print(len(x_advs)) for idx, c_trg in enumerate(c_trg_list): + print(idx) with torch.no_grad(): gen_noattack, gen_noattack_feats = self.G(x_real, c_trg) # Attack