Is log_softmax needed for activations on GPU to use your Transducer Loss? cf https://github.com/HawkAaron/E2E-ASR/blob/04d416b1c32a8cbe55aa7527cfce25739339cbd5/model.py#L83

[cweng6]

[HawkAaron]

No need.

[cweng6]

hmm, but in the latest version of pytorch bindings, it seems only CPU it doesnot log_softmax, see,

https://github.com/HawkAaron/warp-transducer/blob/05c524ef3d8601ed48813b417c820cbdeb023b45/pytorch_binding/warprnnt_pytorch/__init__.py#L67