saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-09-09 21:10:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
stable-baselines3/stable_baselines3/version.txt

2 lines
8 B
Text
Raw Normal View History

Fix memory leak in base_class.py (#1908) * Fix memory leak in base_class.py Loading the data return value is not necessary since it is unused. Loading the data causes a memory leak through the ep_info_buffer variable. I found this while loading a PPO learner from storage on a multi-GPU system since the ep_info_buffer is loaded to the memory location it was on while it was saved to disk, instead of the target loading location, and is then not cleaned up. * Update changelog.rst * Update changelog --------- Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
2024-05-15 13:59:32 +00:00
2.4.0a1
Reference in a new issue Copy permalink