ray/rllib/agents/pg/__init__.py

14 lines
400 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, PGTFPolicy
from ray.rllib.agents.pg.pg_torch_policy import pg_torch_loss, PGTorchPolicy
__all__ = [
"pg_tf_loss",
"pg_torch_loss",
"post_process_advantages",
"DEFAULT_CONFIG",
"PGTFPolicy",
"PGTorchPolicy",
"PGTrainer",
]