deep-learning - 标签变量上的 MXNet 初始化错误

标签 deep-learning mxnet

当我调用 module.fit() 时出现错误 ValueError:labelidx 的未知初始化模式。 符号“labelidx”是我用于标签数据的名称——我不想使用 softmax_label 因为我没有使用 softmax 输出,但这似乎是很多事情。它似乎试图将 labelidx 初始化为参数,这是一个错误。我怎么能告诉它这是输入而不是学习参数?

最佳答案

我想通了。

在构造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',), ...)

关于deep-learning - 标签变量上的 MXNet 初始化错误,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40494602/

相关文章:

python - 在Python中使用MXNet预训练图像分类模型

machine-learning - 在caffe中定义新层时如何获取学习率或迭代次数

python - Ubuntu - 在 python : "ImportError: No module named mxnet" 中运行程序时出错

python-3.x - mxnet 导入 nd 或 np 以使用数组

machine-learning - 如何为我的全连接神经网络找到合适的架构?

python - scikit-learn, keras, tensorflow 和 mxnet 中保存机器学习模型的所有格式是什么?

python - Mxnet模型转换为Onnx模型时不支持Upsampling操作

python - tensorflow 错误 : "Cannot parse tensor from proto"

python - 在 Keras 中制作采样层

python - ValueError : Error when checking input: expected dense_151_input to have 3 dimensions, 但得到形状为 (2, 2100) 的数组