lstm分段输入运行
背景:
拿lstm训练了序列到序列的模型,拟合的还可以,但是考虑到实际运行,整个输入数据不是一下子拿到的,是随时间一段一段拿到,但是输出还要有一定的实时要求,想lstm计算某点时只与前面的点有关和后面的没关系,应该改为分段没啥问题。开搞。
如图模型完成的是 itf -> re_nn 的映射。整段输入没啥,换成分段,果不出其然,是不行的。
寻思着我只是单纯预测,参数并没有传。
参数也传了,总感觉怪怪的,不对啊model.state_dict()
保存的是最优权重和偏置,不应该还改变啊,嗯,应该改的不是这个参数。然后,,然后就卡在这了。
看了看lstm结构,有两个参数是随时间变化参与计算的,直觉告诉我就是他们。
果真,搞定!!!!