GANimation conditional attacks

2020-01-09 12:08:54 -04:00
parent 3cde107dac
commit b0c3723d11
1 changed files with 1 additions and 1 deletions
@@ -420,7 +420,7 @@ class Solver(Utils):
                        # img = regular_image_transform(Image.open(images_to_animate_path[idx])).unsqueeze(0).cuda()

                        # Wrong Class
-                        x_adv, perturb = pgd_attack.perturb(image_to_animate, black, targets[0, :].unsqueeze(0).cuda()
+                        x_adv, perturb = pgd_attack.perturb(image_to_animate, black, targets[0, :].unsqueeze(0).cuda())

                        # Joint Class Conditional
                        # x_adv, perturb = pgd_attack.perturb_joint_class(image_to_animate, black, targets[:, :].cuda())