saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-04 04:07:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
stable-baselines3/stable_baselines3
History
Ole Petersen 6c00565778
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 15:59:32 +02:00
..
a2c Add rollout_buffer_class parameter to on-policy algorithms (#1720) 2023-10-27 17:36:24 +02:00
common Fix memory leak in base_class.py (#1908) 2024-05-15 15:59:32 +02:00
ddpg Update TD3/DDPG/DQN defaults for consistency (#1785) 2024-01-12 16:05:14 +01:00
dqn Update TD3/DDPG/DQN defaults for consistency (#1785) 2024-01-12 16:05:14 +01:00
her Fix various typos (#1926) 2024-05-15 15:19:39 +02:00
ppo Add rollout_buffer_class parameter to on-policy algorithms (#1720) 2023-10-27 17:36:24 +02:00
sac Fix various typos (#1926) 2024-05-15 15:19:39 +02:00
td3 Fix various typos (#1926) 2024-05-15 15:19:39 +02:00
__init__.py Add Gymnasium support (#1327) 2023-04-14 13:13:59 +02:00
py.typed Rename to stable-baselines3 2020-05-05 15:02:35 +02:00
version.txt Fix memory leak in base_class.py (#1908) 2024-05-15 15:59:32 +02:00