diff options
Diffstat (limited to 'include/caffe/database_factory.hpp')
-rw-r--r-- | include/caffe/database_factory.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/include/caffe/database_factory.hpp b/include/caffe/database_factory.hpp index 91185e99..30f191e7 100644 --- a/include/caffe/database_factory.hpp +++ b/include/caffe/database_factory.hpp @@ -9,8 +9,11 @@ namespace caffe { -shared_ptr<Database> DatabaseFactory(const DataParameter_DB& type); -shared_ptr<Database> DatabaseFactory(const string& type); +template <typename K, typename V> +shared_ptr<Database<K, V> > DatabaseFactory(const DataParameter_DB& type); + +template <typename K, typename V> +shared_ptr<Database<K, V> > DatabaseFactory(const string& type); } // namespace caffe |