pytorch nn.Sequential()动态添加方法

最新推荐文章于 2024-06-30 11:35:14 发布

weixin_30905133

最新推荐文章于 2024-06-30 11:35:14 发布

阅读量3.6k

点赞数 3

文章标签：人工智能

原文链接：http://www.cnblogs.com/wanghui-garcia/p/11278061.html

版权

本文介绍了如何使用PyTorch的nn.Sequential()模块动态添加子模块，通过.add_module方法，可以按需添加并用指定的名字访问这些子模块。此功能在构建模型时提供更大的灵活性。

摘要由CSDN通过智能技术生成

之前我们使用nn.Sequential()都是直接写死的，就如下所示：

# 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)),
          ('