美文网首页
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