BCELoss for multilabel / multiclass prediction

h3lio5 / linguistic-style-transfer-pytorch

Implementation of "Disentangled Representation Learning for Non-Parallel Text Style Transfer(ACL 2019)" in Pytorch

68 stars 21 forks source link

Closed PierreColombo closed 4 years ago

PierreColombo commented 4 years ago

Hello, How can you do multiclass, multilabel prediction with BCELoss ?