
一、模型容器——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
网友评论