mirror of
https://github.com/wassname/ray.git
synced 2026-10-29 13:30:42 +08:00
rllib] Fix leak of TensorFlow assign operations in DQN/DDPG
This commit is contained in:
1 parent
09d05bb3fa
commit
34fbc7fb4c
1 file changed
+1
-1
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user