From 65ca9c04fa977a93d6cb298e37d95d4a5b015e6b Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Tue, 24 Dec 2019 12:53:04 -0400 Subject: [PATCH] next --- stargan/solver.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/stargan/solver.py b/stargan/solver.py index f439a90..3db988b 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -635,9 +635,8 @@ 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') + l0_error += np.linalg.norm((gen - gen_noattack).data[0].cpu().numpy(), ord=0) + min_dist += attacks.min_dist((gen - gen_noattack).data[0].cpu().numpy(), ord='-inf') n_samples += 1 # Save the translated images.