2020-04-21 09:47:52 +02:00
|
|
|
# Code in this file is adapted from:
|
|
|
|
|
# https://github.com/openai/evolution-strategies-starter.
|
|
|
|
|
|
|
|
|
|
import ray
|
|
|
|
|
from ray.rllib.agents.es.es_torch_policy import after_init, before_init, \
|
|
|
|
|
make_model_and_action_dist
|
2020-12-26 20:14:18 -05:00
|
|
|
from ray.rllib.policy.policy_template import build_policy_class
|
2020-04-21 09:47:52 +02:00
|
|
|
|
2020-12-26 20:14:18 -05:00
|
|
|
ARSTorchPolicy = build_policy_class(
|
2020-04-21 09:47:52 +02:00
|
|
|
name="ARSTorchPolicy",
|
2020-12-26 20:14:18 -05:00
|
|
|
framework="torch",
|
2020-04-21 09:47:52 +02:00
|
|
|
loss_fn=None,
|
|
|
|
|
get_default_config=lambda: ray.rllib.agents.ars.ars.DEFAULT_CONFIG,
|
|
|
|
|
before_init=before_init,
|
|
|
|
|
after_init=after_init,
|
|
|
|
|
make_model_and_action_dist=make_model_and_action_dist)
|
