This commit is contained in:
Nataniel Ruiz
2019-12-24 14:50:44 -04:00
parent cb4eb8c8fc
commit d63731fb64
+1 -1
View File
@@ -35,7 +35,7 @@ class LinfPGDAttack(object):
y = torch.FloatTensor(y).to(self.device)
self.model.zero_grad()
loss = -self.loss_fn(output, y)
loss = self.loss_fn(output, y)
loss.backward()
grad = X.grad