火炬CTCCriterion错误:您必须将批次中每个序列的大小作为张量传递

时间:2016-11-17 06:59:17

标签: lua deep-learning torch criterion

我计划在Torch中为车牌识别器实施CTC标准。我在warp_ctc中阅读了教程。但是,在培训期间,它失败了,并显示以下错误消息。

/ home / ubuntu / torch / install / bin / luajit:... / ubuntu / torch / install / share / lua / 5.1 / nnx / CTCCriterion.lua:24:你必须传递每个序列的大小批量作为张量 堆栈追溯:         [C]:在函数'断言'中         ... / ubuntu / torch / install / share / lua / 5.1 / nnx / CTCCriterion.lua:24:in function'forward'

ctc标准接收3个参数, CTCCriterion:前锋(输出,标签,大小) this is my target/label, { 1 : CudaTensor - size: 24 2 : CudaTensor - size: 24 3 : CudaTensor - size: 24 4 : CudaTensor - size: 24 5 : CudaTensor - size: 24 6 : CudaTensor - size: 24 7 : CudaTensor - size: 24 8 : CudaTensor - size: 24 9 : CudaTensor - size: 24 10 : CudaTensor - size: 24 }

and output { 1 : CudaTensor - size: 24x36 2 : CudaTensor - size: 24x36 3 : CudaTensor - size: 24x36 4 : CudaTensor - size: 24x36 5 : CudaTensor - size: 24x36 6 : CudaTensor - size: 24x36 7 : CudaTensor - size: 24x36 8 : CudaTensor - size: 24x36 9 : CudaTensor - size: 24x36 10 : CudaTensor - size: 24x36 }

批量化:24,我假设每个完整的车牌组合最多10个字符,36个字母a-z,0-9。

我尝试插入值1,24,36,但在训练期间每轮出现相同的错误消息。我注意到数据类型允许在输出和目标是张量类型而不是表,我尝试转换和重塑。但它仍然没有帮助。

0 个答案:

没有答案