问题描述:
以第8章的rnn为例,其训练代码如下:
model = d2l.RNNModelScratch(len(vocab), num_hiddens, device, get_lstm_params,
init_lstm_state, lstm)
d2l.train_ch8(model, train_iter, vocab, lr, num_epochs, device)
终端上只显示,不显示图像:
Figure size 700x500 with 1 Axes
Figure size 700x500 with 1 Axes
解决办法:
- 进入train_ch8源代码中,找到Animator类里的add方法,加以下代码:
plt.draw()
plt.pause(0.001)
2.训练代码最后加上:
d2l.plt.show()
结果:
动态显示图片
文章讲述了在使用d2l库的RNN模型进行训练时,如何在终端不显示图像的情况下,通过修改train_ch8中的Animator类,添加`plt.draw()`和`plt.pause(0.001)`代码,实现动态显示训练过程中的图片。
4708





