下面咱们来说一下简单神经网络的搭建,以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=self.conv1(x)
x=self.maxpoll1(x)
x=self.conv2(x)
x=self.maxpoll2(x)
x=self.conv3(x)
x=self.maxpoll3(X)
x=self.flatten(x)
x=self.linear1(x)
x=self.linear2(x)
return x
guang=Guang()
print(guang)
运行结果如下:



