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 阅读(1071) 评论(0) 推荐(0) 编辑