CalvinBackup/disrupting-deepfakes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

next

Browse Source
This commit is contained in:
Nataniel Ruiz
2019-12-26 12:09:15 -04:00
parent 5113d9373b
commit e97a04675c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ganimation/solver.py
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, :].cuda())
x_adv, perturb = pgd_attack.perturb_iter_class(image_to_animate, black, targets[target_idx, :].unsqueeze(0).cuda())
x_advs.append(x_adv, perturb)
for target_idx in range(targets.size(0)):