mxnet.io.NDArrayIter(data, label=None, batch_size=1, shuffle=False, last_batch_handle='pad', data_name='data', label_name='softmax_label')
看这里,我们了解到,mxnet.io.NDArrayIter
的label_name
参数默认是softmax_label
。
如果要用其他的,这里必须要修改。改成和symbol中的label一样就行了。