Yixiang Gao 4d54afb6df sparse cat cross entropy (#1597) ...

* add sparse cat cross entropy

* minor fix

* add log_softmax into loss function

* add test

* update docs

* fix training loss

* add device

2023-08-21 14:14:54 -07:00

10 KiB

Executable File

Raw Blame History

View Raw