This commit is contained in:
Nataniel Ruiz
2019-12-23 17:22:38 -04:00
parent 0b90dda881
commit f6bc1c6cb1
2 changed files with 73 additions and 73 deletions
+1 -1
View File
@@ -7,7 +7,7 @@ import torch
import torch.nn as nn
class LinfPGDAttack(object):
def __init__(self, model=None, device=None, epsilon=0.05, k=1, a=0.05, feat = None):
def __init__(self, model=None, device=None, epsilon=0.05, k=20, a=0.01, feat = None):
self.model = model
self.epsilon = epsilon
self.k = k