From 5c17ec0ca23f01650c606d0ac025c87cb505a1c8 Mon Sep 17 00:00:00 2001
From: Nataniel Ruiz <nruiz@Nataniels-MacBook-Pro.local>
Date: Tue, 24 Dec 2019 12:59:39 -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 3db988b..39f9fb5 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[0].cpu().numpy(), ord=0)
-                        min_dist += attacks.min_dist((gen - gen_noattack).data[0].cpu().numpy(), ord='-inf')
+                        l0_error += (gen - gen_noattack).norm(0)
+                        min_dist += (gen - gen_noattack).norm('-inf')
                         n_samples += 1
 
                 # Save the translated images.