浅谈pytorch中的nn.Sequential(*net[3: 5])是啥意思

 更新时间:2021年04月26日 09:06:34   作者:alittlebai1  
这篇文章主要介绍了pytorch中的nn.Sequential(*net[3: 5])是啥意思,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

看到代码里面有这个

在这里插入图片描述

1 class ResNeXt101(nn.Module):
    2 def __init__(self):
        3 super(ResNeXt101, self).__init__()
        4 net = resnext101()
        # print(os.getcwd(), net)

        5 net = list(net.children())  # net.children()得到resneXt 的表层网络
        # for i, value in enumerate(net):
        #     print(i, value)
        6 self.layer0 = nn.Sequential(net[:3])  # 将前三层打包0, 1, 2两层
        print(self.layer0)
        7 self.layer1 = nn.Sequential(*net[3: 5])  # 将3, 4两层打包
        8 self.layer2 = net[5]
        9 self.layer3 = net[6]

可以看到代码中的第六行(序号自己去掉,我打上去的) 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]这个东西为啥要带“ * ”
当代码中不带*的时候,运行会出现以下问题

在这里插入图片描述

意思就是列表不是子类,就是说参数不对

net = list(net.children())

这一行代码是将模型的每一层取出来构建一个列表,自己试着打印就可以。大概的输出就是[conv(),BatchNorm2d(), ReLU,MaxPool2d]等等

在这里插入图片描述

总共是是个元素,和一般的列表不太一样。

当我们取net[:3]的时候,传进去的参数是一个列表,但是我们用*net[:3]的时候传进去的是单个元素

list1 = ["conv", ("relu", "maxing"), ("relu", "maxing", 3), 3]
list2 = [list1[:1]]
list3 = [*list1[:1]]
print("list2:{}, *list1[:2]:{}".format(list1[:1], *list1[:1]))

在这里插入图片描述

结果不带✳的是列表,带✳的是元素,所以nn.Sequential(*net[3: 5])中的*net[3: 5]就是给nn.Sequential()这个容器中传入多个层。

到此这篇关于pytorch中的nn.Sequential(*net[3: 5])是啥意思的文章就介绍到这了,更多相关pytorch nn.Sequential(*net[3: 5])内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Python中if和elif的用法示例及区别详解

    Python中if和elif的用法示例及区别详解

    这篇文章主要给大家介绍了关于Python中if和elif的用法示例及区别的相关资料,除了基本的if和else语句外,elif(即else if)也是一个非常重要的组成部分,需要的朋友可以参考下
    2024-04-04
  • 关于Python 的简单栅格图像边界提取方法

    关于Python 的简单栅格图像边界提取方法

    今天小编就为大家分享一篇关于Python 的简单栅格图像边界提取方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-07-07
  • Python OS模块实例详解

    Python OS模块实例详解

    这篇文章主要介绍了Python OS模块,结合实例形式总结分析了Python使用OS解析文件路径、判断文件、目录等相关操作技巧,需要的朋友可以参考下
    2019-04-04
  • python数学建模之Numpy 应用介绍与Pandas学习

    python数学建模之Numpy 应用介绍与Pandas学习

    这篇文章主要介绍了python数学建模之Numpy 应用介绍与Pandas学习,NumPy 是一个运行速度非常快的数学库,一个开源的的python科学计算库,主要用于数组、矩阵计算
    2022-07-07
  • 手把手带你了解python多进程,多线程

    手把手带你了解python多进程,多线程

    这篇文章主要介绍了python多线程与多进程及其区别详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2021-08-08
  • Java实现的执行python脚本工具类示例【使用jython.jar】

    Java实现的执行python脚本工具类示例【使用jython.jar】

    这篇文章主要介绍了Java实现的执行python脚本工具类,结合实例形式分析了java使用jython.jar执行Python脚本的具体操作技巧,需要的朋友可以参考下
    2018-03-03
  • TensorFlow saver指定变量的存取

    TensorFlow saver指定变量的存取

    这篇文章主要为大家详细介绍了TensorFlow saver指定变量的存取,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-03-03
  • Python入门之模块和包用法详解

    Python入门之模块和包用法详解

    这篇文章主要为大家详细介绍一下Python中的包与模块的使用,文中的示例讲解详细,对我们学习Python有一定帮助,感兴趣的小伙伴可以学习一下
    2022-07-07
  • Python线程之多线程展示详解

    Python线程之多线程展示详解

    这篇文章主要为大家介绍了Python线程之多线程展示,具有一定的参考价值,感兴趣的小伙伴们可以参考一下,希望能够给你带来帮助
    2021-12-12
  • Python决策树分类算法学习

    Python决策树分类算法学习

    这篇文章主要为大家详细介绍了Python决策树分类算法,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2017-12-12

最新评论