From 66ef21ad32570991f3f18ef4674b8590662cc4d5 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Thu, 26 Dec 2019 10:11:02 -0400 Subject: [PATCH] next --- stargan/solver.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/stargan/solver.py b/stargan/solver.py index 86bfdd0..53b00dc 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -614,6 +614,7 @@ class Solver(object): for idx, c_trg in enumerate(c_trg_list): x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg) x_advs.append(x_adv, perturb) + break for idx, c_trg in enumerate(c_trg_list): with torch.no_grad(): @@ -643,6 +644,7 @@ class Solver(object): l0_error += (gen - gen_noattack).norm(0) min_dist += (gen - gen_noattack).norm(float('-inf')) n_samples += 1 + break # Save the translated images. x_concat = torch.cat(x_fake_list, dim=3)