今天进入了新的一课:模型训练
模型训练一般分为四个步骤:
- 构建数据集。
- 定义神经网络模型。
- 定义超参、损失函数及优化器。
- 输入数据集进行训练与评估。
训练,通俗的讲就是给我们的AI做训练,练习。带它看看我们的世界
首先要给模型做训练,就要准备好训练的内容,材料,因此就有了数据集的构建
从数据集 Dataset加载代码,构建数据集
当然,这些东西从一些模型网站上下载即可,图中实例就不改了。
在准备完成“试题"内容之后就要告诉它要考查的内容,到这里就需要去赋予变量
赋予完之后,开始训练!(附上打卡时间)