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

作者:alittlebai1 时间:2023-10-05 05:42:55 

看到代码里面有这个

浅谈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]这个东西为啥要带“ * ”
当代码中不带*的时候,运行会出现以下问题

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

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


net = list(net.children())

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

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

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

当我们取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]))

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

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

来源:https://blog.csdn.net/alittlebai1/article/details/116105471

标签:pytorch,nn.Sequential(,net[3:,5])
0
投稿

猜你喜欢

  • Python标准异常和异常处理详解

    2021-02-23 05:10:36
  • Oracle创建主键自增表(sql语句实现)及触发器应用

    2024-01-17 07:12:37
  • python实发邮件实例详解

    2023-10-13 00:43:01
  • 详解Python3中的迭代器和生成器及其区别

    2022-11-01 00:37:48
  • 关于代码阅读问题的小技巧 脚本之家原创(适合所有网站)不定时更新

    2024-01-04 14:57:42
  • 手写一个python迭代器过程详解

    2021-06-29 07:45:23
  • 值得收藏的10道python 面试题

    2022-04-22 12:10:33
  • 使用Tensorboard工具查看Loss损失率

    2022-04-17 12:50:30
  • python 中collections的 deque使用详解

    2023-02-07 15:43:14
  • django迁移文件migrations的实现

    2022-10-27 02:48:44
  • css学习笔记: 重置默认样式 css reset

    2009-07-19 14:30:00
  • 使用python实现学生信息管理系统

    2022-11-24 21:28:35
  • PyTorch device与cuda.device用法介绍

    2022-07-26 23:49:10
  • SQL Server简单查询示例汇总

    2024-01-27 12:19:41
  • Python在cmd上打印彩色文字实现过程详解

    2022-12-19 07:27:33
  • python 中os模块os.path.exists()的用法说明

    2022-04-30 01:22:26
  • Mysql视图和触发器使用过程

    2024-01-15 22:15:46
  • 详解Python 使用 selenium 进行自动化测试或者协助日常工作

    2023-09-05 03:44:19
  • 详解Golang并发操作中常见的死锁情形

    2023-07-03 13:17:43
  • 如何解决ORA-01843与NLS_DATE_FORMAT问题

    2023-06-30 20:57:14
  • asp之家 网络编程 m.aspxhome.com