PyTorch快速入门教程【小土堆】之Sequential使用和小实战

视频地址神经网络-搭建小实战和Sequential的使用_哔哩哔哩_bilibili

python 复制代码
from torch import nn
from torch.nn import Conv2d, MaxPool2d, Flatten, Linear, Sequential


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=2)
        # self.maxpool2 = MaxPool2d(2)
        # self.conv3 = Conv2d(32, 64, 5, padding=2)
        # self.maxpool3 = MaxPool2d(2)
        # self.flatten = Flatten()
        # self.linear2 = Linear(64, 10)
        # self.linear1 = Linear(1024, 64)

        self.model1 = Sequential(Conv2d(3, 32, 5, padding=2),
                                 MaxPool2d(2),
                                 Conv2d(32, 32, 5, padding=2),
                                 MaxPool2d(2),
                                 Conv2d(32, 64, 5, padding=2),
                                 MaxPool2d(2),
                                 Flatten(),
                                 Linear(1024, 64),
                                 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)
        # x = self.flatten(x)
        # x = self.linear1(x)
        # x = self.linear2(x)
        x = self.model1(x)
        return x


tudui = Tudui()
print(tudui)
相关推荐
凡间客8 小时前
Python编程之常用模块
开发语言·python
景彡先生8 小时前
Python基础语法规范详解:缩进、注释与代码可读性
开发语言·前端·python
说私域8 小时前
从工具理性到价值共生:开源链动2+1模式、AI智能名片与S2B2C商城系统的社会连接重构研究
人工智能·重构·开源
heisd_18 小时前
OpenCV计算机视觉库
人工智能·opencv·计算机视觉
liliangcsdn9 小时前
python如何写数据到excel示例
开发语言·python·excel
CNRio9 小时前
将word和excel快速转换为markdown格式
python·word·excel
小白银子12 小时前
零基础从头教学Linux(Day 52)
linux·运维·服务器·python·python3.11
wb0430720113 小时前
性能优化实战:基于方法执行监控与AI调用链分析
java·人工智能·spring boot·语言模型·性能优化
AAA小肥杨13 小时前
基于k8s的Python的分布式深度学习训练平台搭建简单实践
人工智能·分布式·python·ai·kubernetes·gpu
lichong95115 小时前
Git 检出到HEAD 再修改提交commit 会消失解决方案
java·前端·git·python·github·大前端·大前端++