浅谈pytorch中的nn.Sequential(*net[3: 5])是啥意思
这篇文章主要介绍了pytorch中的nn.Sequential(*net[3: 5])是啥意思,文中通过示例代码介绍的非常详细,对大家的深度学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
这篇文章主要介绍了pytorch中的nn.Sequential(*net[3: 5])是啥意思,文中通过示例代码介绍的非常详细,对大家的深度学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
看到代码里面有这个
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
|
可以看到代码中的第六行(序号自己去掉,我打上去的) self.layer0 = nn.Sequential(net[:3])
和
第七行self.layer1 = nn.Sequential(*net[3: 5])
有一个nn.Sequential(net[:3])
和nn.Sequential(*net[3: 5])
今天不讲nn.Sequential()
用法,意义,作用因为我也不咋明白。惊天就说*net[3: 5]
这个东西为啥要带“ * ”
当代码中不带*
的时候,运行会出现以下问题
意思就是列表不是子类,就是说参数不对
1 |
|
这一行代码是将模型的每一层取出来构建一个列表,自己试着打印就可以。大概的输出就是[conv(),BatchNorm2d(), ReLU,MaxPool2d]
等等
总共是是个元素,和一般的列表不太一样。
当我们取net[:3]
的时候,传进去的参数是一个列表,但是我们用*net[:3]
的时候传进去的是单个元素
1 2 3 4 |
|
结果不带✳的是列表,带✳的是元素,所以nn.Sequential(*net[3: 5])
中的*net[3: 5]
就是给nn.Sequential()
这个容器中传入多个层。
到此这篇关于pytorch中的nn.Sequential(*net[3: 5])是啥意思的文章就介绍到这了,更多相关pytorch nn.Sequential(*net[3: 5])内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
更多推荐
所有评论(0)