使用Sequential搭建一个对CIFAR10的分类模型

最新推荐文章于 2024-01-19 11:22:47 发布

gxyqn626

最新推荐文章于 2024-01-19 11:22:47 发布

阅读量119

点赞数

文章标签：神经网络深度学习人工智能 Powered by 金山文档

本文链接：https://blog.csdn.net/gxyqn626/article/details/129145420

版权

SEQUENTIAL

CLASS torch.nn.Sequential(*args)

# Example of using Sequential
model = nn.Sequential(
          nn.Conv2d(1,20,5),
          nn.ReLU(),
          nn.Conv2d(20,64,5),
          nn.ReLU()
        )

# Example of using Sequential with OrderedDict
model = nn.Sequential(OrderedDict([
          ('conv1', nn.Conv2d(1,20,5)),
          ('relu1', nn.ReLU()),
          ('conv2', nn.Conv2d(20,64,5)),
          ('relu2', nn.ReLU())
        ]))