From 2e12fff97110d8c2462f1cd00949d0e5ee027e89 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 26 Dec 2019 12:49:56 -0400 Subject: [PATCH] next --- cyclegan/test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cyclegan/test.py b/cyclegan/test.py index f44bb60..56522ef 100644 --- a/cyclegan/test.py +++ b/cyclegan/test.py @@ -68,7 +68,9 @@ if __name__ == '__main__': model.set_input(data) # unpack data from data loader with torch.no_grad(): model.forward_noattack() - 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()