From 34fbc7fb4ca188203eb5c8f2292da6e57d06746c Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Wed, 23 Oct 2019 00:28:15 -0700 Subject: [PATCH] rllib] Fix leak of TensorFlow assign operations in DQN/DDPG --- rllib/utils/tf_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rllib/utils/tf_ops.py b/rllib/utils/tf_ops.py index b15f1cf58..8efdd9f1a 100644 --- a/rllib/utils/tf_ops.py +++ b/rllib/utils/tf_ops.py @@ -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: