ch6.8代码不符,lstm中最后一个简单版本实现的最后一个函数train_and_predict_rnn_keras并没有device的参数

[sxzhou1937]

去掉device后报错：

[pengfei-chen]

我也遇到了这个问题，我的tensor版本是 2.2.0。
我在本地的utils.py文件中，第293行，修改代码为
ans_list = self.rnn(X, state)
Y = ans_list[0]
state = ans_list[1:]
保存后，Shutdown jupyter文件，重启章节 6.8 的 jupyter文件 解决了这个问题。
以上，希望能略微帮助到后面的人。

[ChungKingExpress]

我也遇到了这个问题，原函数并不需要device这个变量，我删掉device后，报错信息如下

思考后，我意识到，相比GRU和RNN，LSTM的state除了返回一个隐藏状态以外还返回了一个shape与隐藏状态一样的cell。

测试

我更改了RNNModel类的call函数。如下

回调学习率，再次训练便可以了