From d24879d8a5c0fc29984eb01a15552f6055a055e1 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Fri, 17 Jan 2020 14:40:28 -0500 Subject: [PATCH] blur --- stargan/attacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stargan/attacks.py b/stargan/attacks.py index 8aa59db..30a294e 100644 --- a/stargan/attacks.py +++ b/stargan/attacks.py @@ -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