ray/rllib/agents/ppo/__init__.py

18 lines
579 B
Python

from ray.rllib.agents.ppo.ppo import PPOConfig, PPOTrainer, DEFAULT_CONFIG
from ray.rllib.agents.ppo.ppo_tf_policy import PPOStaticGraphTFPolicy, PPOEagerTFPolicy
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",
"PPOStaticGraphTFPolicy",
"PPOEagerTFPolicy",
"PPOTorchPolicy",
"PPOTrainer",
]