This commit is contained in:
Nataniel Ruiz
2019-12-24 18:59:55 -04:00
parent d63731fb64
commit 7b496aab85

View File

@@ -613,7 +613,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, gen_noattack, c_trg)
x_adv, perturb = pgd_attack.perturb(x_real, x_real, c_trg)
# x_adv = x_real + perturb
# x_adv = self.blur_tensor(x_adv)