会员
周边
捐助
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Js2Hou
博客园
首页
新随笔
联系
订阅
管理
2022年2月9日
`nn.ModuleList`和普通list的区别
摘要: ModuleList是特殊的list,其包含的模块会被自动注册,对所有的Module方法都可见。先给结论:如果要用列表组织模型模块,那么强烈建议使用nn.ModuleList。这有什么好处呢?看下面的例子。 import torch.nn as nn from torchsummary import
阅读全文
posted @ 2022-02-09 12:43 Js2Hou
阅读(1189)
评论(0)
推荐(0)
编辑
公告