gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-23 14:42:31 +01:00
Code Issues Projects Releases Wiki Activity

Revert "Changes to avoid memory leak in rollout worker"

Browse Source 
This reverts commit 801aed5e10.
This commit is contained in:
Gourav Roy
2019-01-02 22:37:12 -08:00
parent 779d3694b4
commit c377363e50
2 changed files with 2 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rl_coach/graph_managers/graph_manager.py
View File

@@ -562,8 +562,7 @@ class GraphManager(object):
screen.warning("No checkpoint to restore in: {}".format(self.task_parameters.checkpoint_restore_dir))
else:
screen.log_title("Loading checkpoint: {}".format(checkpoint.model_checkpoint_path))
if not hasattr(self.agent_params.memory, 'memory_backend_params') or self.agent_params.memory.memory_backend_params.run_type != str(RunType.ROLLOUT_WORKER):
self.checkpoint_saver.restore(self.sess, checkpoint.model_checkpoint_path)
self.checkpoint_saver.restore(self.sess, checkpoint.model_checkpoint_path)
[manager.restore_checkpoint(self.task_parameters.checkpoint_restore_dir) for manager in self.level_managers]