next
This commit is contained in:
1 parent
1b162241ed
commit
0f5f9abbd2
3 files changed
+5
-5
No files matched your search
+1
-1
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user