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