next
This commit is contained in:
1 parent
6689912c98
commit
466d927a7c
2 files changed
+2
-2
No files matched your search
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user