标签变量

时间:2016-11-08 19:06:26

标签: deep-learning mxnet

当我致电module.fit()时,我收到了错误消息 ValueError: Unknown initialization pattern for labelidx。 符号" labelidx"是我用于标签数据的名称 - 我不想使用softmax_label,因为我没有使用softmax输出,但这似乎是很多默认设置的。它似乎试图将labelidx初始化为参数,这是一个错误。我怎么能告诉它这是输入而不是学习参数?

1 个答案:

答案 0 :(得分:3)

我想出来了。

构建Module对象时,需要告诉它数据的名称(data_names)和标签(label_names)。其中每个都应该是一个字符串名称列表。默认情况下data_names=('data',), label_names=('softmax_label',),否则它假定其他所有内容都是学习参数,并会尝试初始化它们,从而导致此错误。文档:http://mxnet.io/api/python/module.html#mxnet.module.module.Module

所以在我的情况下,它需要Module(label_names=('labelidx',), ...)