next
This commit is contained in:
1 parent
7122d93890
commit
e3bf1e8cb5
1 file changed
+1
-1
@@ -7,7 +7,7 @@ import torch
|
|||||||
import torch.nn as nn
|
import torch.nn as nn
|
||||||
|
|
||||||
class LinfPGDAttack(object):
|
class LinfPGDAttack(object):
|
||||||
def __init__(self, model=None, epsilon=0.05, k=20, a=0.01):
|
def __init__(self, model=None, epsilon=0.05, k=1, a=0.05):
|
||||||
self.model = model
|
self.model = model
|
||||||
self.epsilon = epsilon
|
self.epsilon = epsilon
|
||||||
self.k = k
|
self.k = k
|
||||||
|
|||||||
Reference in new issue
Block a user