wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-04 21:04:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add support for tensorflow resource variables (#4438)

Browse Source 
* Adding support for resource variables

Currently resource variable go undetected by the `TensorFlowVariables` since they do not use the same ops for reading values. This change should fix this until a more robust solution is implemented.

* fix varhandle
This commit is contained in:
gehring
2019-03-29 16:23:05 -04:00
committed by Eric Liang
parent 77005d1814
commit 5133b10700
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/experimental/tf_utils.py
+2 -1
View File
@@ -76,7 +76,8 @@ class TensorFlowVariables(object):
if control not in explored_inputs:
queue.append(control)
explored_inputs.add(control)
if "Variable" in tf_obj.node_def.op:
if ("Variable" in tf_obj.node_def.op
or "VarHandle" in tf_obj.node_def.op):
variable_names.append(tf_obj.node_def.name)
self.variables = OrderedDict()
variable_list = [