【文件属性】：
文件名称：对pytorch网络层结构的数组化详解
文件大小：41KB
文件格式：PDF
更新时间：2020-12-24 05:50:47
c OR pytorch
最近再写openpose，它的网络结构是多阶段的网络，所以写网络的时候很想用列表的方式，但是直接使用列表不能将网络中相应的部分放入到cuda中去。
其实这个问题很简单的，使用moduleList就好了。
1 我先是定义了一个函数，用来根据超参数，建立一个基础网络结构
stage = [[3, 3, 3, 1, 1], [7, 7, 7, 7, 7, 1, 1]]
branches_cfg = [[[128, 128, 128, 512, 38], [128, 128, 128, 512, 19]],
[[128, 128, 128, 128, 128, 128, 38], [128,