Closed marcromeyn closed 1 year ago
This PR changes the default loss in BinaryOutput to BCELoss. Currently it's BCEWithLogitsLoss which means we apply sigmoid twice.
BinaryOutput
BCELoss
BCEWithLogitsLoss
https://nvidia-merlin.github.io/models/review/pr-1175
Goals :soccer:
This PR changes the default loss in
BinaryOutput
toBCELoss
. Currently it'sBCEWithLogitsLoss
which means we apply sigmoid twice.