Я использую модель Caff enet. Я успешно выполнил задание по классификации с 9 классами. Затем я попытался изменить его на регрессионную сеть, подготовив другой файл LMDB с метками в диапазоне от 700 до 1400. Я изменил исходный обучающий код и заменил softmax на EuclideanLoss и num_outputs на 1. Я сделал то же самое для тестирования и получил эту ошибку :
"Проверка не пройдена: ExactNumBottomBlobs () == bottom.size () (2 против 1) Слой EuclideanLoss принимает в качестве входных данных 2 нижних BLOB-объекта. * Проверка трассировки стека сбоев: * Прервано (ядро сброшено) ) "Итак, я прокомментировал слой EuclideanLoss:
layer {name:" prob "тип:" EuclideanLoss "bottom:" fc8-cats-dogs-n "top:" prob "}
но теперь я получаю:
Файл "testr.py", строка 86, в pred_probas = out ['prob'] KeyError: 'prob'
Может кто-нибудь помочь мне с этим, пожалуйста?