diff --git a/stargan/solver.py b/stargan/solver.py
index 48b381b..6946bce 100644
--- a/stargan/solver.py
+++ b/stargan/solver.py
@@ -641,10 +641,12 @@ class Solver(object):
 
                 # Save the translated images.
                 x_concat = torch.cat(x_fake_list, dim=3)
-                result_path = os.path.join(self.result_dir, '{}-images.jpg'.format(i+1))
+                # result_path = os.path.join(self.result_dir, '{}-images.jpg'.format(i+1))
+                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:
                     break