pytorch零基础实现手写数学符号识别项目（三）——模型训练与预测
迪丽瓦拉
2024-05-16 06:31:03
0次
模型训练与预测
- 前言
- 评判精度
- 模型训练
- 初始化参数
- 训练和测试
- 调用函数进行训练
- 模型损失
- 模型测试
- 测试函数
- 显示测试结果
前言
在前面的文章中我们重点介绍了数据的形式，数据的加载，并且解决了加载过大的数据集可能导致的OOM问题。同时我们也解决了加载torchvision中现有的模型时遇到的输入和输出端与我们实际问题不匹配的情况，在本篇文章中我们将介绍模型的训练和预测过程。
评判精度
为了评判模型是否足够准确，我们先要定义一个评判精度的函数
def accuracy(y_hat, y):
上一篇：一位粉丝大年初三寄来的字节面经
下一篇：jvm面试题汇总