2.4.2Trainer训练自定义的模型

文档摘要

comments: true title: Trainer训练自定义模型 trainer 前言理论上来讲，基于或实现的模型，并且正确重载对应方法（、等方法），那么就可以借助这个强大的工具避开繁琐的代码编写，提高工作效率。下面将借用动手学深度学习内的线性回归的简洁实现来封装基于实现的模型，以进一步使用。将重点讲解封装步骤，部分不再展开。代码引入函数库在这里人为设定线性回归的真实权重为和，偏置为，接下来的目标是让神经网络无限逼近这个权重。定义数据集定义数据集时，确保重载后的方法返回的字典的对应于模型重载后的方法里面的形参名称。定义模型自定义模型的步骤：重载前向传播逻辑。