如何在 Python (Caffe) 中访问 iter_size 变量？

Question

我正在尝试读取 Python 中 solver.prototxt 文件的参数。我可以使用 solver.iter、solver.param.max_iter 等查看 iter、max_iter、display 等变量。但是，我无法访问 iter_size 参数在更新权重

之前执行 forward/backward 的多次迭代

Answer 1

好像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" 行之后应该可以解决问题。
记得在这个更改后用 make pycaffe.

编译 pycaffe 接口

如何在 Python (Caffe) 中访问 iter_size 变量？

How to access iter_size variable in Python (Caffe)?

python

neural-network

deep-learning

caffe

pycaffe