2015-09-18 1 views
3

Я пытаюсь сохранить модель сети caffe в формате .caffemodel, используя следующий код. Но я получаю ошибку сегментации.Сохранение caffe Model C++ исходный код?

caffe::NetParameter net_param; 
caffe_net_->ToProto(&net_param); 
caffe::WriteProtoToBinaryFile(net_param, caffe_model_path); 

где caffe_net_ - обученная сеть caffe, wgich содержит параметры капли.

+0

'assert (caffe_net_)'? – luk32

+0

Opps. Я не видел, что это годовалый пост. Не уверен, почему он появился в верхней части моего фида. – luk32

ответ

0

Не знаю, почему вы получите отказ от этого. Именно так вы вручную сохраните состояние сети. Вы пробовали пройти через и увидеть, в какой момент это происходит? В противном случае, просто снимите его во время обучения?

+0

Да, я отлаживал программу, и проблема возникает, когда выполняется функция «ToProto». Тем не менее я не могу понять это. Это даст ошибку SIGSEGV –

+0

@cybertrone Я бы порекомендовал шаг за шагом в методе ToProto и определил, где это произошло. –

Смежные вопросы