Support label_smoothing in cross_entropy

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

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.

jbschlosser commented 3 years ago

Closing and reopening diff from the FB side