Fix loss functions - Githubissues

kaylode / theseus

General template for most Pytorch projects

MIT License

34 stars 6 forks source link

Closed kaylode closed 2 years ago

kaylode commented 2 years ago

Fix FocalLoss, SmoothCELoss when using with MixupCutmixCollator. This error happens because these two losses required one-hot encoding.
Resolves #26