torch.py中的train_epoch_ch3函数:

将273行和第275行的l后面加上mean()

image-20220324082242952

torch.py中的class Animator(动态绘图部分):

在其add函数的倒数第二行加入:plt.draw();plt.pause(0.001)

image-20220324082427863