简单的神经网络结构

最新推荐文章于 2024-05-20 23:06:56 发布

爱吃榴莲的妹妹

最新推荐文章于 2024-05-20 23:06:56 发布

阅读量250

点赞数

文章标签：神经网络 pytorch 深度学习

本文链接：https://blog.csdn.net/Crystalxxtt/article/details/124924850

版权

若要构建上图中的神经网络

import torch
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)#通过计算得出padding=2
        self.maxpool1 = MaxPool2d(2)
        self.conv2 = Conv2d(32,32,5,padding = 2)
        self.maxpool2=MaxPool2d(2)
        self.conv3 = Conv2d(32,64,5,padding=2)
        self.maxpool3 = MaxPool2d(2)
        self.flatten = Flatten()
        self.linear1 = Linear(1024,64)
        self.linear2 = Linear(64,10)
        
    
    def forward(self,x):
        x = self.conv1(x)
        x = self.maxpool1(x)
        x = self.conv2(x)
        x = self.maxpool2(x)
        x = self.conv3(x)
        x = self.maxpool3(x)