ray/rllib/agents/pg/__init__.py

9 lines
321 B
Python

from ray.rllib.agents.pg.pg import PGTrainer, DEFAULT_CONFIG
from ray.rllib.agents.pg.pg_tf_policy import pg_tf_loss, \
post_process_advantages
from ray.rllib.agents.pg.pg_torch_policy import pg_torch_loss
__all__ = [
"PGTrainer", "pg_tf_loss", "pg_torch_loss", "post_process_advantages",
"DEFAULT_CONFIG"
]