Datasets
Models
Downloads: 1
Diff of
/loss/BCE.py
[000000]
..
[f77492]
Switch to side-by-side view
--- a +++ b/loss/BCE.py @@ -0,0 +1,9 @@ +import torch.nn as nn +import torch + +loss_function = torch.nn.BCELoss() + +def BCE_loss(input, target): + # input = input.cuda() + # target = target.cuda() + return loss_function(input, target.float().cuda()) \ No newline at end of file