From 2a37e622072aed7cd425694edebe284b7edc19a2 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 26 Dec 2019 13:08:54 -0400 Subject: [PATCH] next --- cyclegan/test.py | 6 +++--- cyclegan/util/attacks.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/cyclegan/test.py b/cyclegan/test.py index d07904c..8ab5082 100644 --- a/cyclegan/test.py +++ b/cyclegan/test.py @@ -70,9 +70,9 @@ if __name__ == '__main__': model.set_input(data) # unpack data from data loader with torch.no_grad(): model.forward_noattack() - # if i == 0: - # input_adv, perturb = model.attack() - input_adv, perturb = model.attack() + if i == 0: + input_adv, perturb = model.attack() + # input_adv, perturb = model.attack() with torch.no_grad(): model.forward_attack(perturb) model.compute_visuals() diff --git a/cyclegan/util/attacks.py b/cyclegan/util/attacks.py index adb3050..a9a654b 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=20, a=0.01): + def __init__(self, model=None, epsilon=0.05, k=1, a=0.05): self.model = model self.epsilon = epsilon self.k = k