songsenand
/
cgal
mirror of https://github.com/CGAL/cgal
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Reduce max memory used

This commit is contained in:
Simon Giraudot 2018-12-06 14:59:26 +01:00
parent 291982ee67
commit e86777c8dc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Classification/include/CGAL/Classification/TensorFlow_neural_network_classifier.h
View File

@ -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;