xla
13597d69 - Nll loss update (#3144)

Commit
4 years ago
Nll loss update (#3144) * When the whole target is equal to ignore_index, return Nan instead of 0 * Add torch pin * AllClose to handle nan in input * Fixed nllloss backward, when target is all ignored, should return 0 instead of nan * Delete .torch_pin
Author
Parents
Loading