From e86777c8dc4a94c2ea3b781ccf6324b6670c2f8e Mon Sep 17 00:00:00 2001 From: Simon Giraudot Date: Thu, 6 Dec 2018 14:59:26 +0100 Subject: [PATCH] Reduce max memory used --- .../Classification/TensorFlow_neural_network_classifier.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Classification/include/CGAL/Classification/TensorFlow_neural_network_classifier.h b/Classification/include/CGAL/Classification/TensorFlow_neural_network_classifier.h index a6b457990ef..307289bd7c2 100644 --- a/Classification/include/CGAL/Classification/TensorFlow_neural_network_classifier.h +++ b/Classification/include/CGAL/Classification/TensorFlow_neural_network_classifier.h @@ -1057,7 +1057,7 @@ void classify (const ItemRange& input, probabilities[i].resize (input.size()); const std::size_t mem_allocated = sizeof(float) * input.size() * (labels.size() + classifier.features().size()); - const std::size_t size_max = 1024 * 1024 * 1024; + const std::size_t size_max = 512 * 1024 * 1024; const std::size_t nb_subdivisions = (mem_allocated / size_max) + 1; std::cerr << nb_subdivisions << " subdivision(s) for GPU processing" << std::endl; @@ -1106,7 +1106,7 @@ void classify (const ItemRange& input, output.resize(input.size()); const std::size_t mem_allocated = sizeof(float) * input.size() * (labels.size() + classifier.features().size()); - const std::size_t size_max = 1024 * 1024 * 1024; + const std::size_t size_max = 512 * 1024 * 1024; const std::size_t nb_subdivisions = (mem_allocated / size_max) + 1; std::cerr << nb_subdivisions << " subdivision(s) for GPU processing" << std::endl;