This commit is contained in:
Nataniel Ruiz
2019-12-30 13:53:13 -04:00
parent 8031d66d71
commit 6dbcf6f93b
+1 -1
View File
@@ -46,7 +46,7 @@ class LinfPGDAttack(object):
self.model.zero_grad()
return X, eta
return X, X - X_nat
def clip_tensor(X, Y, Z):
# Clip X with Y min and Z max