【Pytorch】ModuleList和Sequential和List的区别

  • 文档:torch.nn.ModuleList, Sequential, List
  • ModuleList和List类似,是nn.Module的子类,能够和List一样的索引方式进行所以,但是没有实现forward方法,其中存储的元素自动识别为Module,如果直接用List的话,程序看不到里面的参数之类的东西,训练时会报错。ModuleList can be indexed like a regular Python list, but modules it contains are properly registered, and will be visible by all Module methods.
  • Sequential实现了forward方法,要注意各个module之间的channel要对应起来。

参考:stackflow

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值