This commit is contained in:
Nataniel Ruiz
2019-12-26 09:34:58 -04:00
parent 6689912c98
commit 466d927a7c
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -27,7 +27,7 @@ class LinfPGDAttack(object):
output = self.model(X)
self.model.zero_grad()
loss = -self.loss_fn(output, y)
loss = self.loss_fn(output, y)
loss.backward()
grad = X.grad