Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Juha Reunanen's avatar
    To avoid a GPU memory leak, allow passing thread pools to dnn_trainer from outside (#2027) · 74123841
    Juha Reunanen authored 
    * Problem: The CUDA runtime allocates resources for each thread, and apparently those resources are not freed when the corresponding threads terminate. Therefore, each instantiation of dnn_trainer leaks a bit of GPU memory.

Solution: Add possibility to pass thread pools from outside. This way, subsequent dnn_trainer instances can use the same threads, and there's no memory leak.

* Add helpful comments
    74123841
trainer_abstract.h 36.3 KB