wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-18 12:20:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[RLlib] Curiosity exploration module: tf/tf2.x/tf-eager support. (#11945)

Browse Source
This commit is contained in:
Sven Mika
2020-11-29 12:31:24 +01:00
committed by GitHub
parent a537b852e6
commit fb318addcb
10 changed files with 227 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rllib/execution/multi_gpu_impl.py
+1 -1
View File
@@ -62,7 +62,7 @@ class LocalSyncParallelOptimizer:
self.loss_inputs = input_placeholders + rnn_inputs
self.build_graph = build_graph
# First initialize the shared loss network
# First initialize the shared loss network.
with tf1.name_scope(TOWER_SCOPE_NAME):
self._shared_loss = build_graph(self.loss_inputs)
shared_ops = tf1.get_collection(