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

Support constructing TensorFlowVariables from multiple tf operations (#2182)

Browse Source
This commit is contained in:
Binglin Chang
2018-06-02 18:13:52 -07:00
committed by Robert Nishihara
parent d699bfbf10
commit 19d6ca0670
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/tensorflow_test.py
+1 -1
View File
@@ -128,7 +128,7 @@ class TensorFlowTest(unittest.TestCase):
variables2.set_flat(flat_weights)
assert_almost_equal(flat_weights, variables2.get_flat())
variables3 = ray.experimental.TensorFlowVariables(loss2)
variables3 = ray.experimental.TensorFlowVariables([loss2])
self.assertEqual(variables3.sess, None)
sess = tf.Session()
variables3.set_session(sess)