mirror of
https://github.com/vale981/ray
synced 2025-03-07 02:51:39 -05:00
3e6ba5d6d2
* Revert "Revert "[RLlib] POC: `PGTrainer` class that works by sub-classing, not `trainer_template.py`. (#20055)" (#20284)"
This reverts commit 246787cdd9.
Co-authored-by: sven1977 <svenmika1977@gmail.com>
14 lines
431 B
Python
14 lines
431 B
Python
from ray.rllib.agents.pg.pg import PGTrainer, DEFAULT_CONFIG
|
|
from ray.rllib.agents.pg.pg_tf_policy import pg_tf_loss, PGTFPolicy
|
|
from ray.rllib.agents.pg.pg_torch_policy import pg_torch_loss, PGTorchPolicy
|
|
from ray.rllib.agents.pg.utils import post_process_advantages
|
|
|
|
__all__ = [
|
|
"pg_tf_loss",
|
|
"pg_torch_loss",
|
|
"post_process_advantages",
|
|
"DEFAULT_CONFIG",
|
|
"PGTFPolicy",
|
|
"PGTorchPolicy",
|
|
"PGTrainer",
|
|
]
|