mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -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.
17 lines
397 B
YAML
17 lines
397 B
YAML
cartpole-ars-torch:
|
|
env: CartPole-v0
|
|
run: ARS
|
|
stop:
|
|
episode_reward_mean: 150
|
|
timesteps_total: 500000
|
|
config:
|
|
use_pytorch: true
|
|
noise_stdev: 0.02
|
|
num_rollouts: 50
|
|
rollouts_used: 25
|
|
num_workers: 2
|
|
sgd_stepsize: 0.01
|
|
noise_size: 25000000
|
|
eval_prob: 0.5
|
|
model:
|
|
fcnet_hiddens: [64, 64]
|