1. 程式人生 > >解決mxnet報錯:KeyError: 'softmax_label' 迭代器預設的label_name='softmax_label'。

解決mxnet報錯:KeyError: 'softmax_label' 迭代器預設的label_name='softmax_label'。

mxnet.io.NDArrayIter(data, label=None, batch_size=1, shuffle=False, last_batch_handle='pad', data_name='data', label_name='softmax_label')

  看這裡,我們瞭解到,mxnet.io.NDArrayIterlabel_name引數預設是softmax_label
  如果要用其他的,這裡必須要修改。改成和symbol中的label一樣就行了。