美文网首页
pytorch【模型模块】二、模型容器与AlexNet构建

pytorch【模型模块】二、模型容器与AlexNet构建

作者: 米线织毛衣 | 来源:发表于2020-02-18 22:01 被阅读0次

一、模型容器——containers

ModuleList可以让网络层像list那样迭代,ModuleDict可以给每个网络层命名并索引。

1.容器之sequential

(代码见lesson-11)

2.容器之ModuleList

3.容器之ModuleDict

4.三者的对比与总结

二、AlexNet

相关文章

网友评论

      本文标题:pytorch【模型模块】二、模型容器与AlexNet构建

      本文链接:https://www.haomeiwen.com/subject/rxgnfhtx.html