mirror of
https://github.com/wassname/ray.git
synced 2026-09-12 12:51:15 +08:00
[RLlib] Add testing framework_iterator. (#7852)
* Add testing framework_iterator. * LINT. * WIP. * Fix and LINT. * LINT fix.
This commit is contained in:
@@ -135,7 +135,7 @@ def fc(x, weights, biases=None):
|
||||
isinstance(weights, torch.Tensor) else weights
|
||||
biases = biases.detach().numpy() if \
|
||||
isinstance(biases, torch.Tensor) else biases
|
||||
if tf:
|
||||
if tf and tf.executing_eagerly():
|
||||
x = x.numpy() if isinstance(x, tf.Variable) else x
|
||||
weights = weights.numpy() if isinstance(weights, tf.Variable) else \
|
||||
weights
|
||||
|
||||
Reference in New Issue
Block a user