diff --git a/labml_nn/gan/cycle_gan.py b/labml_nn/gan/cycle_gan.py
index 7baab3f3705eede2d58bf77dd37c84a9ae7f4987..63488958a454c13d915d5a11ee5101d6a4ca19a5 100644
--- a/labml_nn/gan/cycle_gan.py
+++ b/labml_nn/gan/cycle_gan.py
@@ -493,7 +493,7 @@ class Configs(BaseConfigs):
         # Loop through epochs
         for epoch in monit.loop(self.epochs):
             # Loop through the dataset
-            for i, batch in enumerate(self.dataloader):
+            for i, batch in monit.enum('Train', self.dataloader):
                 # Move images to the device
                 data_x, data_y = batch['x'].to(self.device), batch['y'].to(self.device)
 
@@ -528,6 +528,8 @@ class Configs(BaseConfigs):
             # Update learning rates
             self.generator_lr_scheduler.step()
             self.discriminator_lr_scheduler.step()
+            # New line
+            tracker.new_line()
 
     def optimize_generators(self, data_x: torch.Tensor, data_y: torch.Tensor, true_labels: torch.Tensor):
         """
diff --git a/setup.py b/setup.py
index 1d7b3bee01421ce749447a1c2b8ac10c24fc634b..68e604bf8ebfd2bc0ddd6100cb1dc732134af8ba 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@ with open("readme.md", "r") as f:
 
 setuptools.setup(
     name='labml-nn',
-    version='0.4.80',
+    version='0.4.81',
     author="Varuna Jayasiri, Nipun Wijerathne",
     author_email="vpjayasiri@gmail.com, hnipun@gmail.com",
     description="A collection of PyTorch implementations of neural network architectures and layers.",