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

Change tf_utils.py get_weights to evaluate all tensors at once rather than calling tensor.eval per-tensor. (#8491)

Browse source
This commit is contained in:
internetcoffeephone 2020-05-19 07:06:03 +02:00 • committed by GitHub
parent 6c5ea32857
commit a73c488c74
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
python/ray/experimental/tf_utils.py
View file

@ -161,10 +161,7 @@ class TensorFlowVariables:
Dictionary mapping variable names to their weights.
"""
self._check_sess()
return {
k: v.eval(session=self.sess)
for k, v in self.variables.items()
}
return self.sess.run(self.variables)
def set_weights(self, new_weights):
"""Sets the weights to new_weights.