Ошибка при построении набора данных для модели цепочки вручную

Возникает ошибка, когда я обучаю NN с ручным набором данных. Мой код находится здесь. Запустите 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, проблем не будет. Я не знаю почему, кто-нибудь понял? Заранее спасибо.


person jian chang    schedule 17.05.2018    source источник


Ответы (1)


Значение сообщения об ошибке: тип dtype аргумента №1 (0-происхождение, т. е. метка) для softmax_cross_entropy будет i (целое число со знаком), но задано u (целое число без знака). Попробуйте привести его, например, mnist.train.labels.astype(numpy.int32).

person kmaehashi    schedule 17.05.2018