diff options
Diffstat (limited to 'python/caffe/_caffe.cpp')
| -rw-r--r-- | python/caffe/_caffe.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp index 0e38dee7..8687dd87 100644 --- a/python/caffe/_caffe.cpp +++ b/python/caffe/_caffe.cpp @@ -134,8 +134,8 @@ void Net_SetInputArrays(Net<Dtype>* net, bp::object data_obj, Solver<Dtype>* GetSolverFromFile(const string& filename) { SolverParameter param; - ReadProtoFromTextFileOrDie(filename, ¶m); - return GetSolver<Dtype>(param); + ReadSolverParamsFromTextFileOrDie(filename, ¶m); + return SolverRegistry<Dtype>::CreateSolver(param); } struct NdarrayConverterGenerator { |