ligmap options weren't actually implemented - would rather they are

specified in the model file
gnina · Jan 24, 2018 · 5485c31 · 5485c31
1 parent 5d74321
commit 5485c31
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/include/caffe/solver.hpp b/include/caffe/solver.hpp
@@ -65,7 +65,7 @@ class Solver {
   // that stores the learned net. You should implement the SnapshotSolverState()
   // function that produces a SolverState protocol buffer that needs to be
   // written to disk together with the learned net.
-  void Snapshot();
+  string Snapshot();
   virtual ~Solver() {}
   inline const SolverParameter& param() const { return param_; }
   inline shared_ptr<Net<Dtype> > net() { return net_; }

diff --git a/src/caffe/solver.cpp b/src/caffe/solver.cpp
@@ -401,7 +401,7 @@ void Solver<Dtype>::Test(const int test_net_id) {
 }
 
 template <typename Dtype>
-void Solver<Dtype>::Snapshot() {
+string Solver<Dtype>::Snapshot() {
   CHECK(Caffe::root_solver());
   string model_filename;
   switch (param_.snapshot_format()) {
@@ -416,6 +416,7 @@ void Solver<Dtype>::Snapshot() {
   }
 
   SnapshotSolverState(model_filename);
+  return model_filename;
 }
 
 template <typename Dtype>