diff --git a/stargan/solver.py b/stargan/solver.py
index 1e98f63..b7a7798 100644
--- a/stargan/solver.py
+++ b/stargan/solver.py
@@ -731,10 +731,10 @@ class Solver(object):
                 # result_path = os.path.join(self.result_dir, '{}-{}-images.jpg'.format(layer_num_orig, i+1))
                 save_image(self.denorm(x_concat.data.cpu()), result_path, nrow=1, padding=0)
                 # print('Saved real and fake images into {}...'.format(result_path))
-                # if i == 3:
-                #     break
-                if i == 199:
+                if i == 3:
                     break
+                # if i == 199:
+                #     break
             
             # Print metrics
             print('{} images. L1 error: {}. L2 error: {}. L0 error: {}. L_-inf error: {}. Perceptual error: {}.'.format(n_samples,