现在我们有了模型和数据,是时候通过优化数据参数来训练、验证和测试我们的模型了。训练一个模型就是一个迭代的过程,在每次迭代(称为epoch)中,模型对输出进行猜测,计算猜测中的误差(损失),收集误差对其参数的导数(前一节中自动微分一样),并使用梯度下降优化这些参数。
1、先决条件代码
从前面关于数据集和数据装载器以及构建模型的章节中加载代码。
PyTorch基础-自定义数据集和数据加载器(2)_一只小小的土拨鼠的博客-CSDN博客
pytorch基础-构建简单的神经网络(4)_一只小小的土拨鼠的博客-CSDN博客
import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from t