Support label_smoothing for cross_entropy in nestedtensor

pytorch / nestedtensor

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.

BSD 3-Clause "New" or "Revised" License

252 stars 28 forks source link

Closed jbschlosser closed 3 years ago

jbschlosser commented 3 years ago

Summary: Adds awareness of the new label_smoothing functionality in cross_entropy added in https://github.com/pytorch/pytorch/pull/63122 to nestedtensor's implementation.

Fixes broken test: test.test_nested_tensor_functional.TestFunctional.

Differential Revision: D30730728

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D30730728

jbschlosser commented 3 years ago

Closing because this merged I guess