just in case, better handling of Caffe db cursor reset

revilokeb · Mar 9, 2015 · e301216 · e301216
1 parent bb3e145
commit e301216
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/caffeinputconns.cc b/src/caffeinputconns.cc
@@ -372,7 +372,9 @@ namespace dd
   void ImgCaffeInputFileConn::reset_dv_test()
   {
     //_test_db = std::unique_ptr<caffe::db::DB>();
-    _test_db_cursor = std::unique_ptr<caffe::db::Cursor>();
+    if (_test_db)
+      _test_db_cursor = std::unique_ptr<db::Cursor>(_test_db->NewCursor());
+    else _test_db_cursor = std::unique_ptr<caffe::db::Cursor>();
   }
 
 }