文件名称：对pytorch网络层结构的数组化详解
文件大小：41KB
文件格式：PDF
更新时间：2024-01-13 23:37:27
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,