Show what to do next

models/meta_arch/binary_seg.py

@@ -245,6 +245,7 @@ def losses(self, predictions, targets):
             align_corners=False,
         )
 
+        # TODO BCE cannot handle ignore values, need to implement a custom loss
         loss = F.binary_cross_entropy_with_logits(predictions, targets.float(), reduction="mean")
         losses = {"loss_bce_seg": loss * self.loss_weight}
         return losses