From 6fdd442e20a5f25fe417fb3169254ef499cb4207 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Tue, 31 Dec 2019 22:27:38 -0400 Subject: [PATCH] next --- ganimation/solver.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ganimation/solver.py b/ganimation/solver.py index 1e9ac73..310426f 100644 --- a/ganimation/solver.py +++ b/ganimation/solver.py @@ -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(