hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[RLlib] Fix "tf variable is unhashable" Error. (#24273)

Browse source
This commit is contained in:
HJasperson 2022-04-29 03:07:02 -05:00 • committed by GitHub
parent 2c0f9d7e8f
commit 5f12c62226
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
rllib/models/catalog.py
View file

@ -515,7 +515,7 @@ class ModelCatalog:
def track_var_creation(next_creator, **kw): def track_var_creation(next_creator, **kw):
v = next_creator(**kw) v = next_creator(**kw)
created.add(v) created.add(v.ref())
return v return v
with tf.variable_creator_scope(track_var_creation): with tf.variable_creator_scope(track_var_creation):