CIFAR10
文章平均质量分 78
MarkAssassin
西电本科在读
展开
-
pytorch模型验证(以CIFAR10数据集为例)
在搭建并训练好了自己的神经网络模型后该怎样验证呢?验证模型的方法是,将一张图片进行修改后导入到训练好的模型,通过输出的tensor值与CIFAR10数据集中自带的序列表进行比对。原创 2023-01-03 17:20:50 · 656 阅读 · 0 评论 -
pytorch 模型训练(以CIFAR10数据集为例)
本文简要介绍了pytorch模型训练的一个基本流程,并以CIFAR10数据集进行了演示。但这种方法是在CPU(device=“cpu”)上进行训练的,训练速度比较慢,如果数据集十分庞大不建议使用这种方法,应该在GPU(device=“cuda”)上进行训练。原创 2022-12-20 12:10:18 · 3592 阅读 · 4 评论