This commit is contained in:
Nataniel Ruiz
2019-12-31 22:27:38 -04:00
parent df8731f80e
commit 6fdd442e20
+2 -1
View File
@@ -424,7 +424,8 @@ class Solver(Utils):
targets_au = targets[target_idx, :].unsqueeze(0).cuda()
# x_adv, perturb = pgd_attack.perturb(image_to_animate, black, targets_au)
x_adv, perturb = x_advs[target_idx]
x_adv = image_to_animate + perturb
# x_adv = image_to_animate + perturb
x_adv = image_to_animate
# print(image_to_animate.shape, x_adv.shape)
with torch.no_grad():
resulting_images_att, resulting_images_reg = self.G(