mirror of
https://github.com/wassname/ray.git
synced 2026-08-11 11:24:51 +08:00
[RLlib] Implement PPO torch version. (#6826)
This commit is contained in:
@@ -537,7 +537,7 @@ You can use the ``with_updates`` method on Trainers and Policy objects built wit
|
||||
.. code-block:: python
|
||||
|
||||
from ray.rllib.agents.ppo import PPOTrainer
|
||||
from ray.rllib.agents.ppo.ppo_policy import PPOTFPolicy
|
||||
from ray.rllib.agents.ppo.ppo_tf_policy import PPOTFPolicy
|
||||
|
||||
CustomPolicy = PPOTFPolicy.with_updates(
|
||||
name="MyCustomPPOTFPolicy",
|
||||
|
||||
Reference in New Issue
Block a user