CIFAR 10 model 结构

最新推荐文章于 2024-06-09 15:47:03 发布

qq_45991003

最新推荐文章于 2024-06-09 15:47:03 发布

阅读量879

点赞数

分类专栏： PyTorch深度学习文章标签：神经网络深度学习 cnn python virtualenv

本文链接：https://blog.csdn.net/qq_45991003/article/details/125210734

版权

本文介绍了CIFAR10数据集上的简单神经网络搭建过程。模型结构包括卷积层、最大池化层，详细描述了网络的每一步操作，如32@32x32的卷积、最大池化至32@16x16等，最终通过展开层得到输出结果。代码实现和运行效果一并展示。

摘要由CSDN通过智能技术生成

下面咱们来说一下简单神经网络的搭建，以CIFAR10 结构及简要叙述，只搭建神经网络，数据集在官网下载。

首先要经过一个卷积，32@32x32，接着最大池化，32@16x16，再卷积32@16x16，再最大池化，32@8x8，最后卷64@8x8，最后最大池化，64@4x4，接着展开，最后输出结果。

废话不说，上代码，

class Guang(nn.Module):
    def __init__(self):
        super(Tudui, self).__init__()
        self.conv1=Conv2d(3,32,5,padding=2)
        self.maxpoll1=MaxPool2d(2)
        self.conv2=Conv2d(32,32,5,padding=2)
        self.maxpoll2=MaxPool2d(2)
        self.conv3=Conv2d(32,64,5,padding=2)
        self.maxpoll3=MaxPool2d(2)
        self.flatten=Flatten()
        self.linear1=Linear(1024,64)
        self.linear2=Linear(64,10)
    def forward(self,x):
        x=