From d3dc2c60fdeeeaa9c4df6adcceb0bf2dfdaf8095 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 26 Dec 2019 12:52:43 -0400 Subject: [PATCH] next --- cyclegan/test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cyclegan/test.py b/cyclegan/test.py index 56522ef..95af62e 100644 --- a/cyclegan/test.py +++ b/cyclegan/test.py @@ -68,9 +68,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()