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.