gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-14 10:32:32 +02: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 committed 2019-01-02 23:09:09 -08:00
1 parent 779d3694b4
commit c377363e50
2 files changed
+2 -27
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

rl_coach/graph_managers/graph_manager.py
+1 -2
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]