深度学习框架搭建
文章平均质量分 72
weixin_51431157
这个作者很懒,什么都没留下…
展开
-
Resnet18实现CIFAR10数据集图像分类
Resnet18结构如下:可以看到,18层的网络有五个部分组成,从conv2开始,每层都有两个有残差块,并且每个残差块具有2个卷积层。其中,蓝色部分为conv2,然后往下依次按颜色划分为conv3、conv4,conv5。需要注意的是,从conv3开始,第一个残差块的第一个卷积层的stride为2,这是每层图片尺寸变化的原因。原创 2023-03-29 19:17:22 · 1102 阅读 · 0 评论 -
VGG16实现CIFAR10数据集
为了防止在梯度下降过程中出现梯度消失或爆炸,还需在每一层卷积后使用归一化处理batchnorm,也可防止过拟合,全连接层和线性层后使用dropout随机失活防止过拟合;此外还需加上非线性激活函数Relu。由tensorboard中的曲线趋势可见,提高epoch轮次,正确率还可进一步提升。简单神经网络实现CIFAR10图片分类效果较差,故将网络结构换成VGG16看看效果。更换model.py文件里的内容(1中博客的model),将Links改为VGG16。比起原代码多了数据增强的步骤(transform)原创 2023-03-28 19:38:28 · 756 阅读 · 0 评论 -
CIFAR10模型训练任务
pytorch学习原创 2023-03-22 22:01:57 · 1434 阅读 · 2 评论 -
深度学习框架介绍
深度学习框架本质可以看作一个库(或包),是一个简单的写满了函数声明的py文件,使用户(调包侠)可以通过调用其中的函数轻松完成深度学习模型(神经网络)的创建和训练工作(不用另外定义函数公式,直接输入值,得到输出值)。原创 2023-03-14 16:21:13 · 466 阅读 · 0 评论