ray/rllib/tuned_examples/regression_tests/cartpole-impala-tf.yaml
Sven Mika 166bb5d690
[RLlib] IMPALA PyTorch (#8287)
This PR adds an IMPALA PyTorch implementation.

- adds compilation tests for LSTM and w/o LSTM.
- adds learning test for CartPole.
2020-05-03 13:44:25 +02:00

9 lines
191 B
YAML

cartpole-impala-tf:
env: CartPole-v0
run: IMPALA
stop:
episode_reward_mean: 150
timesteps_total: 500000
config:
use_pytorch: false
num_gpus: 0