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