我正在尝试在Python中读取solver.prototxt文件的参数。我可以使用itermax_iter等查看诸如displaysolver.itersolver.param.max_iter等的变量。但是,在更新权重之前,我无法访问执行向前/向后多次迭代的iter_size参数。

最佳答案

似乎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_iterdisplay是公开的,但iter_size没有公开。
我想加一行

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


"display"行之后应该可以解决问题。
记住在使用make pycaffe进行此更改后,编译pycaffe接口。

关于python - 如何在Python(Caffe)中访问iter_size变量?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/41776188/

10-09 04:29