From 4edd2bb07c73563bbdf8b4282e6f08e78a9ec679 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 26 Dec 2019 12:27:35 -0400 Subject: [PATCH] next --- ganimation/solver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ganimation/solver.py b/ganimation/solver.py index 2cce867..1e9ac73 100644 --- a/ganimation/solver.py +++ b/ganimation/solver.py @@ -411,7 +411,7 @@ class Solver(Utils): if idx == 0: for target_idx in range(targets.size(0)): - x_adv, perturb = pgd_attack.perturb_iter_class(image_to_animate, black, targets[target_idx, :].unsqueeze(0).cuda()) + x_adv, perturb = pgd_attack.perturb(image_to_animate, black, targets[target_idx, :].unsqueeze(0).cuda()) x_advs.append((x_adv, perturb)) for target_idx in range(targets.size(0)):