This commit is contained in:
Nataniel Ruiz
2019-12-26 12:27:35 -04:00
parent 91747e4033
commit 4edd2bb07c

View File

@@ -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)):