Невозможно скопировать веса параметров 0 из слоя 'conv1 / 7x7_s2 / bn';несоответствие формы.Caffe Error - PullRequest
0 голосов
/ 24 мая 2018

Я обучил предварительно обученный GoogleNet (caffe) по Цифрам.Обучение прошло хорошо, и тестирование в цифровом интерфейсе не показало никаких проблем.Но когда я экспортирую модель и пытаюсь использовать example.py, то цифры предоставляются.Эта ошибка возникла.

Невозможно скопировать веса параметров 0 из слоя 'conv1 / 7x7_s2 / bn';несоответствие формы h.Форма исходного параметра: 1 64 1 1 (64);Целевой параметр формы составляет 64 (64).Чтобы узнать параметры этого слоя из sc ratch, а не копировать из сохраненной сети, переименуйте слой.

Я проверил deploy.prototxt и train_val.prototxt.Размер правильный.

Есть предложения?

1 Ответ

0 голосов
/ 24 мая 2018

Попробуйте отредактировать 'deploy.prototxt' для слоя 'conv1/7x7_s2/bn'.Добавьте

param {
  share_mode: PERMISSIVE
}

Для каждого из обучаемых параметров слоя.

См. caffe.proto для получения дополнительной информации.

...