神经网络——CIFAR10小实战

绿树疯子

于 2024-08-28 08:45:32 发布

阅读量529

点赞数 2

分类专栏：机器学习文章标签：神经网络人工智能深度学习 pytorch

本文链接：https://blog.youkuaiyun.com/qq_37441377/article/details/141618931

版权

1.引子

在这里插入图片描述

Sequential的使用：将网络结构放入其中即可，可以简化代码。
找了一个对CIFAR10进行分类的模型。

2.代码实战

from torch import nn
from torch.nn import Conv2d, MaxPool2d, Flatten, Linear

class Tudui(nn.Module):
    def __init__(self):
        super(Tudui, self).__init__()
        self.conv1 = Conv2d(3, 32, 5, padding=2)
        self.maxpool1 = MaxPool2d(2)
        self.conv2 = Conv2d(32, 32, 5, padding=