rllib] Fix leak of TensorFlow assign operations in DQN/DDPG

This commit is contained in:
Eric Liang
2019-10-23 00:28:15 -07:00
committed by GitHub
parent 09d05bb3fa
commit 34fbc7fb4c
+1 -1
View File
@@ -70,7 +70,7 @@ def make_tf_callable(session_or_none, dynamic_shape=False):
else:
args_flat.append(a)
args = args_flat
if not placeholders:
if symbolic_out[0] is None:
with session_or_none.graph.as_default():
for i, v in enumerate(args):
if dynamic_shape: