可能由于d2l版本原因,导致d2l中没有train_epoch_ch3()训练函数,具体情况如下:
为了检测softmax以及mlp训练效果,本人将前面知识整合,编写了一个python文件,以便进行训练检验,并与原先代码相比,增加了测试集损失函数曲线随训练轮数的变化曲线,具体操作如下:
1、 下载hrl.py文件,下载路径:
git clone git@github.com:HenryLiukkk/hrl.git
2、clone成功后,会对应出现一个文件夹,如下图所示:
3、将hrl.py放入李沐深度学习课程文件中,如下图所示:
4、库导入时加一行import hrl:
5、使用hrl(对照原先代码进行修改):
num_epochs, lr = 10, 0.1
updater = torch.optim.SGD(params, lr=lr)
hrl_1 = hrl.train_epoch_ch3_hrl(net, train_iter, test_iter, loss, num_epochs, updater)
hrl_1.main()
hrl.predict_ch3(net, test_iter,n=6)
6、效果展示: