CalvinBackup/disrupting-deepfakes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

next

Browse Source
This commit is contained in:
Nataniel Ruiz
2019-12-23 16:49:26 -04:00
parent 1b162241ed
commit 0f5f9abbd2
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
stargan/attacks.py
+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)