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

GANimation conditional attacks

Browse Source
This commit is contained in:
Nataniel Ruiz
2020-01-09 14:13:17 -04:00
parent 969027540c
commit f659e2af19
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ganimation/attacks.py
+6 -6
View File
@@ -30,8 +30,8 @@ class LinfPGDAttack(object):
out = imFromAttReg(output_att, output_img, X)
self.model.zero_grad()
# loss = -self.loss_fn(output_att, y) + self.loss_fn(output_img, y)
loss = -self.loss_fn(out, y)
loss = -self.loss_fn(output_att, y)
# loss = -self.loss_fn(out, y)
loss.backward()
grad = X.grad
@@ -96,8 +96,8 @@ class LinfPGDAttack(object):
self.model.zero_grad()
# loss = self.loss_fn(output_att, y)
loss = -self.loss_fn(out, y)
loss = -self.loss_fn(output_att, y)
# loss = -self.loss_fn(out, y)
loss.backward()
grad = X.grad
@@ -131,8 +131,8 @@ class LinfPGDAttack(object):
out = imFromAttReg(output_att, output_img, X)
# loss = -self.loss_fn(output_att, y)
loss = -self.loss_fn(out, y)
loss = -self.loss_fn(output_att, y)
# loss = -self.loss_fn(out, y)
full_loss += loss
full_loss.backward()