diff options
| author | Ronghang Hu <huronghang@hotmail.com> | 2015-09-24 19:40:45 -0700 |
|---|---|---|
| committer | Ronghang Hu <huronghang@hotmail.com> | 2015-10-16 22:32:32 -0700 |
| commit | 0eea815ad6fa3313888b6229499a237820258deb (patch) | |
| tree | b76664a58993a0b0658a01f24dd3d0f8898ceac1 /tools | |
| parent | b822a702d19d4fbebbc91198a991f91c34e60650 (diff) | |
| download | caffeonacl-0eea815ad6fa3313888b6229499a237820258deb.tar.gz caffeonacl-0eea815ad6fa3313888b6229499a237820258deb.tar.bz2 caffeonacl-0eea815ad6fa3313888b6229499a237820258deb.zip | |
Change solver type to string and provide solver registry
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/caffe.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tools/caffe.cpp b/tools/caffe.cpp index e3f684b5..1cb6ad89 100644 --- a/tools/caffe.cpp +++ b/tools/caffe.cpp @@ -194,7 +194,7 @@ int train() { GetRequestedAction(FLAGS_sighup_effect)); shared_ptr<caffe::Solver<float> > - solver(caffe::GetSolver<float>(solver_param)); + solver(caffe::SolverRegistry<float>::CreateSolver(solver_param)); solver->SetActionFunction(signal_handler.GetActionFunction()); |