pytorch add_module

最新推荐文章于 2024-01-23 09:44:52 发布

不解不惑

最新推荐文章于 2024-01-23 09:44:52 发布

阅读量1k

点赞数 1

分类专栏：深度学习文章标签： pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/qq_24815615/article/details/125686997

版权

深度学习专栏收录该内容

27 篇文章 2 订阅

订阅专栏

在使用pytorch进行深度学习开发的时候，在一个模型的基础上进行模块的添加，可以使用add_module，可以采用nn.Sequential，也可以采用nn.Module。

（1）nn.Sequential()

import torch.nn as nn
 
model = nn.Sequential()
 
model.add_module("conv1", nn.Conv2d(1, 20, 5))
model.add_module('relu1', nn.ReLU())
model.add_module('conv2', nn.Conv2d(20, 64, 5))
model.add_module('relu2', nn.ReLU())

(2) nn.Module()

# 被添加的module可以通过 name 属性来获取。
 
import torch.nn as nn
class Model(nn.Module):
    def __init__(self):
        super(Model, self).__init__()
        self.add_module("conv", nn.Conv2d(10, 20, 4))
        # self.conv = nn.Conv2d(10, 20, 4) 和上面这个增加module的方式等价
model = Model()
print(model.conv) # 通过name属性访问添加的子模块
print(model)

参考文献：

1.用torch.nn.Sequential()搭建神经网络模型