diff --git a/pix2pixHD_attack/util/attacks.py b/pix2pixHD_attack/util/attacks.py index b6296ff..d43c61c 100644 --- a/pix2pixHD_attack/util/attacks.py +++ b/pix2pixHD_attack/util/attacks.py @@ -7,7 +7,7 @@ import torch import torch.nn as nn class LinfPGDAttack(object): - def __init__(self, model=None, epsilon=0.05, k=1, a=0.05): + def __init__(self, model=None, epsilon=0.05, k=20, a=0.01): self.model = model self.epsilon = epsilon self.k = k