This commit is contained in:
Nataniel Ruiz committed 2019-12-23 16:49:26 -04:00
1 parent 1b162241ed
commit 0f5f9abbd2
3 files changed
+5 -5

No files matched your search

+1 -1
View File
@@ -29,7 +29,7 @@ class LinfPGDAttack(object):
output, feats = self.model(X, c_trg)
if self.feat:
print('self.feat ', self.feat)
# print('self.feat ', self.feat)
output = feats[self.feat]
y = np.zeros(output.shape)
y = torch.FloatTensor(y).to(self.device)