我正在实验Watson神经网络建模器。
我已经从内置演示“ MNIST上的单卷积层”创建了一个模型。我所做的唯一自定义操作是指定训练数据文件。
然后我导出了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用户/专家来阐明此问题?我基本上是在运行演示。它不应该失败。
谢谢!