gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-01-07 22:34:23 +01:00
Code Issues Projects Releases Wiki Activity

Revert "Avoid Memory Leak in Rollout worker"

Browse Source 
This reverts commit c694766fad.
This commit is contained in:
Gourav Roy
2019-01-02 22:35:06 -08:00
parent 2461892c9e
commit 6dd7ae2343
3 changed files with 11 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rl_coach/agents/agent.py
View File

@@ -953,12 +953,6 @@ class Agent(AgentInterface):
self.input_filter.restore_state_from_checkpoint(checkpoint_dir, checkpoint_prefix)
self.pre_network_filter.restore_state_from_checkpoint(checkpoint_dir, checkpoint_prefix)
if self.ap.task_parameters.framework_type == Frameworks.tensorflow:
import tensorflow as tf
tf.reset_default_graph()
# Recreate all the networks of the agent
self.networks = self.create_networks()
# no output filters currently have an internal state to restore
# self.output_filter.restore_state_from_checkpoint(checkpoint_dir)