2017-01-21 8 views
2

Я пытаюсь прочитать параметры файла solver.prototxt в Python. Я могу видеть, как переменные iter, max_iter, display и т.д. с помощью solver.iter, solver.param.max_iter и т.д. Однако, я не могу получить доступ к параметру iter_size, который выполняет несколько итераций вперед/назад перед обновлением весовКак получить доступ к переменной iter_size в Python (Caffe)?

ответ

3

Похоже, что iter_size не подвергается pycaffe interface:

bp::class_<SolverParameter>("SolverParameter", bp::no_init) 
    .add_property("max_iter", &SolverParameter::max_iter) 
    .add_property("display", &SolverParameter::display) 
    .add_property("layer_wise_reduce", &SolverParameter::layer_wise_reduce); 

Как вы можете видеть max_iter и display подвергаются, но не iter_size.
Я полагаю, добавив строку

.add_property("iter_size", &SolverParameter::iter_size) 

После "display" линии следует сделать трюк.
Не забудьте скомпилировать интерфейс pycaffe после этого изменения с помощью make pycaffe.