深度学习的训练、预测过程详解【以LeNet模型和CIFAR10数据集为例】
Lenet是一个 7 层的神经网络(不包含输入层),包含 3 个卷积层,2 个池化层,2 个全连接层。
使用pytorch搭建如下:
下载地址:https://tensorflow.google.cn/datasets/catalog/cifar10
CIFAR10数据集共有60000张彩色图像,其中50000张用于训练,5个训练批,每一批10000张图;10000张用于测试。
图片大小为3X32X32,分为10个类别,每个类6000张。
对于模型的训练可以分为一下几个步骤:下面就结合代码进行详细分析:..
转载
2022-07-08 09:33:01 ·
974 阅读 ·
1 评论