mirror of
https://github.com/vale981/ray
synced 2025-03-08 11:31:40 -05:00
166bb5d690
This PR adds an IMPALA PyTorch implementation. - adds compilation tests for LSTM and w/o LSTM. - adds learning test for CartPole.
9 lines
193 B
YAML
9 lines
193 B
YAML
cartpole-impala-torch:
|
|
env: CartPole-v0
|
|
run: IMPALA
|
|
stop:
|
|
episode_reward_mean: 150
|
|
timesteps_total: 500000
|
|
config:
|
|
use_pytorch: true
|
|
num_gpus: 0
|