pycaffe中SGDSolver和get_solver方法有什么区别?

时间:2016-10-27 05:21:30

标签: python caffe pycaffe

我正在尝试在pycaffe上实现DNN(mnist)。

pretrained_solver = caffe.SGDSolver('mnist/lenet_solver.prototxt')

pretrained_solver = caffe.get_solver('mnist/lenet_solver.prototxt')

上面两行有什么区别? 似乎我用任何一行得到了相同的结果......

1 个答案:

答案 0 :(得分:1)

根据https://github.com/BVLC/caffe/issues/3033

  

第一个(get_solver)允许您指定解算器类型   prototxt。

     

第二个(SGDSolver)不太可靠,因为如果指定了   原型中不同的求解器类型,它仍将使用   SGDSolver并没有告诉你,你有不匹配。