手动为chainer模型构建数据集时出错

时间:2018-05-17 04:28:16

标签: neural-network dataset chainer

使用手动数据集训练NN时出错。我的代码是here。运行 python train_with_manual_data.py --manual 1 ,输出错误是:

InvalidType: 
Invalid operation is performed in: SoftmaxCrossEntropy (Forward)

Expect: in_types[1].dtype.kind == i
Actual: u != i

但是运行 python train_with_manual_data.py --manual 0 ,就没有问题了。我不知道为什么,任何人都能得到线索?提前谢谢。

1 个答案:

答案 0 :(得分:2)

错误消息的含义是:dtype kind的参数#1(0-origin,即label)预计为i(有符号整数),但是给出u(无符号整数)。 尝试投射它,例如mnist.train.labels.astype(numpy.int32)