From 46e85e45c67f715115d5c4b30f6b7ddcb04d092d Mon Sep 17 00:00:00 2001 From: HarveyBrezinaConniffe Date: Tue, 13 Feb 2018 21:28:46 +0000 Subject: [PATCH] Added set_dnn_prefer_smallest_algorithms() (#1128) * Added set_dnn_prefer_smallest_algorithms() * Added description for set_dnn_prefer_smallest_algorithms();w * Changed architectures to algorithms. --- tools/python/src/cnn_face_detector.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/python/src/cnn_face_detector.cpp b/tools/python/src/cnn_face_detector.cpp index 94fd1e35a..ad7ced5cc 100644 --- a/tools/python/src/cnn_face_detector.cpp +++ b/tools/python/src/cnn_face_detector.cpp @@ -155,6 +155,7 @@ void bind_cnn_face_detection(py::module& m) "takes a list of images as input returning a 2d list of mmod rectangles" ); } + m.def("set_dnn_prefer_smallest_algorithms", &set_dnn_prefer_smallest_algorithms, "Tells cuDNN to use slower algorithms that use less RAM."); { typedef mmod_rect type; py::class_(m, "mmod_rectangle", "Wrapper around a rectangle object and a detection confidence score.")