facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification
https://classyvision.ai
MIT License
1.59k stars 278 forks source link

Fix classy losses test #780

Closed jdsgomes closed 2 years ago

jdsgomes commented 2 years ago

Summary: Classivision test_cross_entropy_loss due to a change in behaviour in torch (see 31116297)

Differential Revision: D34424419

facebook-github-bot commented 2 years ago

@jdsgomes has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.