【发布时间】:2018-09-12 14:58:15
【问题描述】:
我有一个定义的模型和定义的层,我将定义层的 n 个实例添加到模型的 init 函数中的列表中,如下所示:
self.layers = []
for i in range(len(nhid)-1):
self.layers.append(MyLayer(nhid[i], nhid[i+1]))
但是当我创建优化器时
optim.Adam(model.parameters(),lr=args.lr, weight_decay=args.weight_decay)
上面写着:
ValueError: 优化器得到一个空的参数列表
但是当我将它写成如下两层时,我没有错误:
self.layer1 = MyLayer[nhid[0],nhid[1]]
self.layer2 = MyLayer[nhid[1],nhid[2]]
【问题讨论】:
标签: python python-3.x pytorch