mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00
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.
10 lines
284 B
Python
10 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",
|
|
]
|