This commit is contained in:
Nataniel Ruiz
2019-12-30 13:46:34 -04:00
parent ce23031e8e
commit c99ef56150
+4 -4
View File
@@ -610,8 +610,8 @@ class Solver(object):
x_fake_list = [x_real]
# x_advs = []
if i == 0:
x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg_list[0])
# if i == 0:
# x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg_list[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))
@@ -621,9 +621,9 @@ 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 = x_real + perturb
# x_adv = self.blur_tensor(x_adv)
# Metrics