一、模型容器——containers
ModuleList可以让网络层像list那样迭代,ModuleDict可以给每个网络层命名并索引。
1.容器之sequential
(代码见lesson-11)
2.容器之ModuleList
3.容器之ModuleDict
4.三者的对比与总结
二、AlexNet
ModuleList可以让网络层像list那样迭代,ModuleDict可以给每个网络层命名并索引。
本文标题:pytorch【模型模块】二、模型容器与AlexNet构建
本文链接:https://www.haomeiwen.com/subject/rxgnfhtx.html
网友评论