From 649a39d3d2f71cb73c37b7ceb5be67c64f40ec52 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 26 Dec 2019 13:02:07 -0400 Subject: [PATCH] next --- cyclegan/util/attacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cyclegan/util/attacks.py b/cyclegan/util/attacks.py index a9a654b..adb3050 100644 --- a/cyclegan/util/attacks.py +++ b/cyclegan/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