mirror of
https://github.com/wassname/ray.git
synced 2026-06-29 04:44:28 +08:00
Sven Mika
d15609ba2a
This PR implements a PyTorch version of RLlib's ARS algorithm using RLlib's functional algo builder API. It also adds a regression test for ARS (torch) on CartPole.
11 lines
284 B
Python
11 lines
284 B
Python
from ray.rllib.agents.ars.ars import ARSTrainer, DEFAULT_CONFIG
|
|
from ray.rllib.agents.ars.ars_tf_policy import ARSTFPolicy
|
|
from ray.rllib.agents.ars.ars_torch_policy import ARSTorchPolicy
|
|
|
|
__all__ = [
|
|
"ARSTFPolicy",
|
|
"ARSTorchPolicy",
|
|
"ARSTrainer",
|
|
"DEFAULT_CONFIG",
|
|
]
|