Skip to content

Commit 9366e2d

Browse files
committed
add custom bce unweighted to config
1 parent 5c82d21 commit 9366e2d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

configs/loss/bce_unweighted.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
class_path: torch.nn.BCEWithLogitsLoss
1+
class_path: chebai.loss.bce_weighted.UnWeightedBCEWithLogitsLoss

0 commit comments

Comments
 (0)