oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard

ctc_loss support long input

Open marigoold opened this issue 2 years ago • 0 comments

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

  • ctc_loss共有四个输入参数,除log_prob之外,其余都是整数形式。实现过程中发现torch支持三个整数参数 int32/64 任意类型的组合,而oneflow只能同时为int64或int32。故增加了两个模板参数,和torch的接口对齐。

文档: image image

marigoold avatar Aug 09 '22 09:08 marigoold