Pytorch 号传参用法 -- nn.Sequential(layers)

客官°小女子只卖身不卖艺 2022-09-10 07:25 295阅读 0赞

\*作用在形参上，代表这个位置接收任意多个非关键字参数，转化成元组方式；\*作用在实参上，代表的是将输入迭代器拆成一个个元素。

# nn.Sequential #

一个有序的容器，神经网络模块将按照在传入构造器的顺序依次被添加到计算图中执行，同时以神经网络模块为元素的有序字典也可以作为传入参数。

举例（手动创建的VGGNet网络的 init 函数。 self.features 和 self.classifier）：

def __init__(self, num_classes = 1000):
            super(VGG, self).__init__()
            layers = []
            in_dim = 3
            out_dim = 64
            # 循环结构卷积层，一共有13个卷积层
            for i in range(13):
                layers += [nn.Conv2d(in_dim, out_dim, 3, 1, 1), nn.ReLU(inplace=True)]
                in_dim = out_dim
                # 在第2、4、7、10、13个卷积层后增加池化层
                if i == 1 or i == 3 or i == 6 or i == 9 or i == 12:
                    layers += [nn.MaxPool2d(2, 2)]
                    # 在第10个卷积后，保持和前面的通道数一致，都为512，其余加倍
                    if i != 9:
                        out_dim *= 2
    
            self.features = nn.Sequential(*layers)
            # VGGNet的3个全连接层，中间有 ReLU 与 Dropout 层
            self.classifier = nn.Sequential(
                nn.Linear(512 * 7 * 7, 4096),
                nn.ReLU(True),
                nn.Dropout(),
                nn.Linear(4096, 4096),
                nn.ReLU(True),
                nn.Dropout(),
                nn.Linear(4096, num_classes),
            )

发表评论取消回复

表情：

评论列表（有 0 条评论，295人围观）

还没有评论，来说两句吧...

相关阅读

相关 C++ | 引用传参，如何传参

前言 > 在定义函数时函数括号中的变量名成为`形式参数`，简称`形参或虚拟参数`；在主调函数中调用一个函数时，该函数括号中的参数名称为`实际参数`，简称实参，实参可以是`

- 日理万妓/ 2022年09月12日 03:50/ 0 赞/ 358 阅读

相关 Pytorch 号传参用法 -- nn.Sequential(layers)

\作用在形参上，代表这个位置接收任意多个非关键字参数，转化成元组方式；\作用在实参上，代表的是将输入迭代器拆成一个个元素。 nn.Sequential 一个有序的容器，

客官°小女子只卖身不卖艺/ 2022年09月10日 07:25/ 0 赞/ 296 阅读

相关传参操作

void Swap1(int a, int b) { int temp; temp = a;

浅浅的花香味﹌/ 2022年08月04日 01:47/ 0 赞/ 243 阅读

相关游标传参

declare -- 定义变量 row_info varchar2(200); -- 定义游标,并指定参数类型

Myth丶恋晨/ 2022年07月26日 05:19/ 0 赞/ 266 阅读

相关 ionic 传参

.state('prepareRegister', { url: '/prepare_register', params:{ 'po

野性酷女/ 2022年07月14日 14:27/ 0 赞/ 231 阅读

相关 vuex 传参

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0hlcm9f

ゝ一纸荒年。/ 2022年03月28日 14:42/ 0 赞/ 415 阅读

相关 C++引用方式传参，传值方式传参（*、&）

C++引用方式传参，传值方式传参（\、&）和数组作为参数传递 1、引用方式传参 2、传值方式传参

「爱情、让人受尽委屈。」/ 2022年02月22日 15:04/ 0 赞/ 450 阅读

相关传参问题

前几天通过action?begin=2013-11-27方式传参，结合struts，结果到action中获得的参数变成了'2013-11-27,2013-11-27'，太奇怪了

迈不过友情╰/ 2022年01月17日 02:39/ 0 赞/ 385 阅读

相关 vue传参之params传参query传参

vue编程式导航，在这里我们必须要明白一个知识点，就是params和query传参的用法。 params传参，路径不能使用path，只能使用name，不然取

我就是我/ 2021年09月16日 02:42/ 0 赞/ 626 阅读

相关 rn 路由传参、navigation传参

路由传参: (1)传参数 (1)发送函数式传递 function xx({navigation

傷城~/ 2021年07月24日 16:57/ 0 赞/ 1333 阅读