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.")