-
Notifications
You must be signed in to change notification settings - Fork 2
Open
Labels
Description
Hi! Thanks for this cool project. I think it'd be very useful to add in support for multilabel classification as well. This might require training with BCEWithLogitsLoss, and accepting a different dataset format. I attempted to add it in but ran into some bugs. If interested, this feature would be appreciated. If I manage to get it working, I'll send a PR.