把数据送到网络,通过前向传播得到预测值
预测值和标签值作比较,根据标签值和预测值的差别 计算损失函数
根据损失值,利用反向传播算法来更新网络参数 达到机器学习的目的
1、如何送数据,继承pytorch Dataset
2、如何定义网络结构 继承pytorch的nn.Module
3、一个epoch代表所有的训练集,每个batch乘个数等于总的训练个数
把数据送到网络,通过前向传播得到预测值
预测值和标签值作比较,根据标签值和预测值的差别 计算损失函数
根据损失值,利用反向传播算法来更新网络参数 达到机器学习的目的
1、如何送数据,继承pytorch Dataset
2、如何定义网络结构 继承pytorch的nn.Module
3、一个epoch代表所有的训练集,每个batch乘个数等于总的训练个数