ray/rllib/agents/ppo/__init__.py
2022-05-22 13:05:24 +02:00

17 lines
515 B
Python

from ray.rllib.agents.ppo.ppo import PPOConfig, PPOTrainer, DEFAULT_CONFIG
from ray.rllib.agents.ppo.ppo_tf_policy import PPOTFPolicy
from ray.rllib.agents.ppo.ppo_torch_policy import PPOTorchPolicy
from ray.rllib.agents.ppo.appo import APPOConfig, APPOTrainer
from ray.rllib.agents.ppo.ddppo import DDPPOConfig, DDPPOTrainer
__all__ = [
"APPOConfig",
"APPOTrainer",
"DDPPOConfig",
"DDPPOTrainer",
"DEFAULT_CONFIG",
"PPOConfig",
"PPOTFPolicy",
"PPOTorchPolicy",
"PPOTrainer",
]