ray/rllib/agents/es/__init__.py
Sven Mika 3812bfedda
[RLlib] PyTorch version of ES (Evolution Strategies). (#8104)
PyTorch version of Evolution Strategies (ES) Algo.
2020-04-20 21:47:28 +02:00

5 lines
253 B
Python

from ray.rllib.agents.es.es import ESTrainer, DEFAULT_CONFIG
from ray.rllib.agents.es.es_tf_policy import ESTFPolicy
from ray.rllib.agents.es.es_torch_policy import ESTorchPolicy
__all__ = ["ESTFPolicy", "ESTorchPolicy", "ESTrainer", "DEFAULT_CONFIG"]