This commit is contained in:
Nataniel Ruiz
2019-12-26 11:26:51 -04:00
parent 22fa209b26
commit 9d6354e879
+6 -6
View File
@@ -610,17 +610,17 @@ class Solver(object):
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
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 = self.blur_tensor(x_adv)