Watson生成的Pytorch结果为:“ ValueError:优化器获得了空的参数列表”

时间:2018-08-08 12:30:26

标签: python code-generation conv-neural-network pytorch ibm-watson

我正在实验Watson神经网络建模器。

我已经从内置演示“ MNIST上的单卷积层”创建了一个模型。我所做的唯一自定义操作是指定训练数据文件。

enter image description here

然后我导出了Pytorch代码,并尝试在本地计算机上运行。

生成的代码可读性强。相关的代码摘录为:

# Define network architecture
class Net(nn.Module):
    def __init__(self, inp_c):
        super(Net, self).__init__()

    def forward(self, ImageData_4, target):

        Convolution2D_9 = self.Convolution2D_9(ImageData_4)
        ReLU_1 = self.ReLU_1(Convolution2D_9)
        Pooling2D_8 = self.Pooling2D_8(ReLU_1)
        Flatten_2 = Pooling2D_8.view(-1, 10816)
        Dense_3 = self.Dense_3(Flatten_2)
        Softmax_5 = self.Softmax_5(Dense_3)
        Accuracy_6 = torch.topk(Softmax_5, 1)[0]
        CrossEntropyLoss_7 = self.CrossEntropyLoss_7(Softmax_5, target)
        return Softmax_5, Accuracy_6

# Model Initialization
inp_c = 1
model = Net(inp_c)
model.cuda()

# Define optimizer
learning_rate = 0.001000
decay = 0.000000
beta_1 = 0.900000
beta_2 = 0.999000
optim = optim.Adam(
    model.parameters(),
    lr=learning_rate,
    betas=(beta_1, beta_2),
    weight_decay=decay)

我遇到了错误:

"ValueError: optimizer got an empty parameter list" 

optim = optim.Adam()语句上。

那边是否有Watson用户/专家来阐明此问题?我基本上是在运行演示。它不应该失败。

谢谢!

0 个答案:

没有答案