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

2020-02-18  本文已影响0人  米线织毛衣

一、模型容器——containers

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

1.容器之sequential

(代码见lesson-11)

2.容器之ModuleList

3.容器之ModuleDict

4.三者的对比与总结

二、AlexNet

上一篇下一篇

猜你喜欢

热点阅读