bce logits loss as same as deepgo

Author: aditya0by0

chebai_proteins/loss/bce_logits_loss.py

@@ -0,0 +1,7 @@
+import torch
+
+
+class WrappedBCEWithLogitsLoss(torch.nn.BCEWithLogitsLoss):
+    def forward(self, input, target, **kwargs):
+        # As the custom passed kwargs are not used in BCEWithLogitsLoss, we can ignore them
+        return super().forward(input, target)

configs/loss/BCELoss.yml

@@ -0,0 +1 @@
+class_path: chebai_proteins.loss.bce_logits_loss.WrappedBCEWithLogitsLoss