问题描述:
以第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()

结果:
动态显示图片
