why network need caption_lengths as an input parameter?

zhenming33 / RAN_torch

radical analysis network

Apache License 2.0

8 stars 5 forks source link

Closed SweetPotatoo closed 5 years ago

SweetPotatoo commented 5 years ago

请问是否需要将label的长度输入网络，不是需要预测的吗？谢谢

zhenming33 commented 5 years ago

caption length是用来控制训练的seqnence的长度的，超过这个长度的没有必要计算。predict的时候这个参数设为所有label的最大长度就可以了。label的结尾由标识符预测。

SweetPotatoo commented 5 years ago

caption length是用来控制训练的seqnence的长度的，超过这个长度的没有必要计算。predict的时候这个参数设为所有label的最大长度就可以了。label的结尾由标识符预测。

感谢解答