Skip to content

Master fix memory leak

Sebastien Rettie requested to merge srettie/umami:master-fix-memory-leak into master

When running on GPU, I observe a memory leak (see orange line in attached image) due to the way tensorflow is storing the validation data for the callback function. This MR fixes the memory leak based on this. It would be great if others could confirm they observe a similar memory leak when running on GPUs before merging.

Also in this MR: bugfix for ttbar test files and tracks test samples.

GPU_Memory_Usage

Edited by Sebastien Rettie

Merge request reports