小土堆：Pytorch深度学习：搭建小实战和Sequential

最近忙着写论文，今天更新Pytorch深度学习：搭建小实战和Sequential。

import torchfrom tensorboardX import SummaryWriterfrom torch import nnfrom torch.nn import Conv2d, MaxPool2d, Flatten, Linear, Sequential
class Yang(nn.Module):    def __init__(self):        super(Yang, self).__init__()        # Sequential模块可以按照顺序添加其他的模块        self.model1 = Sequential(            # Conv2d是卷积层，输入为3通道的图像，输出为32个特征图，卷积核大小为5x5，padding为2            Conv2d(3, 32, 5, padding=2),            # MaxPool2d是池化层，将特征图尺寸缩小一半            MaxPool2d(2),            Conv2d(32, 32, 5, padding=2),            MaxPool2d(2),            Conv2d(32, 64, 5, padding=2),            MaxPool2d(2),            # Flatten是将多维的输入压缩成一维的输出            Flatte