added some notes and removed more dead code

Author: dmarx

src/pytti/LossAug/DepthLossClass.py

@@ -29,7 +29,6 @@ def init_AdaBins(device=None):
 class DepthLoss(MSELoss):
     @torch.no_grad()
     def set_comp(self, pil_image):
-        # pil_image = pil_image.resize(self.image_shape, Image.LANCZOS)
         self.comp.set_(DepthLoss.make_comp(pil_image))
         if self.use_mask and self.mask.shape[-2:] != self.comp.shape[-2:]:
             self.mask.set_(TF.resize(self.mask, self.comp.shape[-2:]))

src/pytti/LossAug/LatentLossClass.py

@@ -25,6 +25,7 @@ def __init__(
             TF.resize(comp.clone(), (h, w)), weight, stop, name, image_shape
         )
 
+    # Comp and mask should live on the image representation, not the loss class.
     @torch.no_grad()
     def set_comp(self, pil_image, device=DEVICE):
         self.pil_image = pil_image

src/pytti/LossAug/MSELossClass.py

@@ -58,6 +58,8 @@ def set_mask(self, mask, inverted=False, device=None):
     def convert_input(cls, input, img):
         return input
 
+    # Comp and mask should live on the image representation, not the loss class.
+    # comp for sure
     @classmethod
     def make_comp(cls, pil_image, device=None):
         if device is None: