mirror of
https://github.com/wassname/ray.git
synced 2026-09-10 12:38:43 +08:00
[rllib] Allow access to batches prior to postprocessing (#4871)
This commit is contained in:
@@ -101,6 +101,10 @@ Tuned examples: `PongNoFrameskip-v4 <https://github.com/ray-project/ray/blob/mas
|
||||
|
||||
**APPO-specific configs** (see also `common configs <rllib-training.html#common-parameters>`__):
|
||||
|
||||
.. warning::
|
||||
|
||||
Keras custom models are not compatible with multi-GPU (this includes PPO in single-GPU mode). This is because the multi-GPU implementation in RLlib relies on variable scopes to implement cross-GPU support.
|
||||
|
||||
.. literalinclude:: ../../python/ray/rllib/agents/ppo/appo.py
|
||||
:language: python
|
||||
:start-after: __sphinx_doc_begin__
|
||||
|
||||
Reference in New Issue
Block a user