diff --git a/codes/models/pixel_level_contrastive_learning/pixpro_lucidrains.py b/codes/models/pixel_level_contrastive_learning/pixpro_lucidrains.py
index a40be66c..73a0a973 100644
--- a/codes/models/pixel_level_contrastive_learning/pixpro_lucidrains.py
+++ b/codes/models/pixel_level_contrastive_learning/pixpro_lucidrains.py
@@ -489,9 +489,7 @@ class PixelCL(nn.Module):
 
         pix_loss = (loss_pixpro_one_two + loss_pixpro_two_one) / 2
 
-        # total loss
-        loss = pix_loss * self.alpha + instance_loss
-        return loss, positive_pixel_pairs
+        return instance_loss, pix_loss, positive_pixel_pairs
 
     # Allows visualizing what the augmentor is up to.
     def visual_dbg(self, step, path):