From b9ad79d66f8dc01afad87ae4a43aa53e65b4395a Mon Sep 17 00:00:00 2001 From: Sven Mika Date: Sun, 2 Feb 2020 02:41:38 +0100 Subject: [PATCH] Add cartpole PPO torch to regression (besides tf). (#7005) --- .../regression_tests/cartpole-ppo-torch.yaml | 11 +++++++++++ 1 file changed, 11 insertions(+) create mode 100644 rllib/tuned_examples/regression_tests/cartpole-ppo-torch.yaml diff --git a/rllib/tuned_examples/regression_tests/cartpole-ppo-torch.yaml b/rllib/tuned_examples/regression_tests/cartpole-ppo-torch.yaml new file mode 100644 index 000000000..3d3ad4291 --- /dev/null +++ b/rllib/tuned_examples/regression_tests/cartpole-ppo-torch.yaml @@ -0,0 +1,11 @@ +cartpole-ppo-tf: + env: CartPole-v0 + run: PPO + stop: + episode_reward_mean: 150 + timesteps_total: 100000 + config: + num_workers: 1 + batch_mode: complete_episodes + observation_filter: MeanStdFilter + use_pytorch: true