wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-16 11:27:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move TensorFlowVariables to ray.experimental.tf_utils. (#4145)

Browse Source
This commit is contained in:
Robert Nishihara
2019-02-24 14:26:46 -08:00
committed by Philipp Moritz
parent 615d5516d1
commit 7b04ed059e
14 changed files with 181 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/source/using-ray-with-tensorflow.rst
+7 -6
View File
@@ -54,8 +54,8 @@ method.
.. code-block:: python
import ray
variables = ray.experimental.TensorFlowVariables(loss, sess)
import ray.experimental.tf_utils
variables = ray.experimental.tf_utils.TensorFlowVariables(loss, sess)
The ``TensorFlowVariables`` object provides methods for getting and setting the
weights as well as collecting all of the variables in the model.
@@ -96,6 +96,7 @@ complex Python objects.
import tensorflow as tf
import numpy as np
import ray
import ray.experimental.tf_utils
ray.init()
@@ -123,7 +124,7 @@ complex Python objects.
init = tf.global_variables_initializer()
self.sess = tf.Session()
# Additional code for setting and getting the weights
self.variables = ray.experimental.TensorFlowVariables(self.loss, self.sess)
self.variables = ray.experimental.tf_utils.TensorFlowVariables(self.loss, self.sess)
# Return all of the data needed to use the network.
self.sess.run(init)
@@ -254,7 +255,7 @@ For reference, the full code is below:
init = tf.global_variables_initializer()
self.sess = tf.Session()
# Additional code for setting and getting the weights
self.variables = ray.experimental.TensorFlowVariables(self.loss, self.sess)
self.variables = ray.experimental.tf_utils.TensorFlowVariables(self.loss, self.sess)
# Return all of the data needed to use the network.
self.sess.run(init)
@@ -320,7 +321,7 @@ For reference, the full code is below:
if iteration % 20 == 0:
print("Iteration {}: weights are {}".format(iteration, weights))
.. autoclass:: ray.experimental.TensorFlowVariables
.. autoclass:: ray.experimental.tf_utils.TensorFlowVariables
:members:
Troubleshooting
@@ -346,7 +347,7 @@ class definiton ``Network`` with a ``TensorFlowVariables`` instance:
sess = tf.Session()
init = tf.global_variables_initializer()
sess.run(init)
self.variables = ray.experimental.TensorFlowVariables(c, sess)
self.variables = ray.experimental.tf_utils.TensorFlowVariables(c, sess)
def set_weights(self, weights):
self.variables.set_weights(weights)