oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard
verified publisher icon Oneflow-Inc

[WIP] ctc_loss support long input

Open marigoold opened this issue 3 years ago • 0 comments

https://github.com/Oneflow-Inc/oneflow/pull/8888 的另一种实现

背景：https://github.com/Oneflow-Inc/OneCloud/issues/104#issuecomment-1156122806 概述：oneflow的ctc_loss不接受int64的target输入而torch能 实现：

  • 直接在functor层cast，而不是注册kernel
  • 删除了max_target_lengths参数，在functor里面计算

marigoold avatar Aug 21 '22 16:08 marigoold