From f665283dfaa0d273350a83f443ab733d408a6d98 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Tue, 24 Dec 2019 12:51:35 -0400 Subject: [PATCH] next --- stargan/solver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stargan/solver.py b/stargan/solver.py index 25e4b25..79e6340 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -635,8 +635,8 @@ class Solver(object): l1_error += F.l1_loss(gen, gen_noattack) l2_error += F.mse_loss(gen, gen_noattack) - l0_error += np.linalg.norm((gen - gen_noattack).data.cpu(), ord=0) - min_dist += attacks.min_dist((gen - gen_noattack).data.cpu(), ord='-inf') + 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 # Save the translated images.