From 4f81836324493eab1a46cf4380f41a4759b12abd Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Tue, 24 Dec 2019 12:52:17 -0400 Subject: [PATCH] next --- stargan/solver.py | 1 + 1 file changed, 1 insertion(+) diff --git a/stargan/solver.py b/stargan/solver.py index 79e6340..f439a90 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -635,6 +635,7 @@ class Solver(object): l1_error += F.l1_loss(gen, gen_noattack) l2_error += F.mse_loss(gen, gen_noattack) + print((gen - gen_noattack).data[0].cpu(), (gen - gen_noattack).data[0].cpu().shape) l0_error += np.linalg.norm((gen - gen_noattack).data[0].cpu(), ord=0) min_dist += attacks.min_dist((gen - gen_noattack).data[0].cpu(), ord='-inf') n_samples += 1