[RLlib] Add PG torch regression test (#6828)

* Add PG torch regression test to tuned_examples/regression_tests dir.

* Rename cartpole-pg.yaml into cartpole-pg-tf.yaml

* cartpole-pg-tf.yaml: Change cartpole-pg name of tuned_example to cartpole-pg-tf.
This commit is contained in:
Sven Mika
2020-01-18 15:57:12 -08:00
committed by Eric Liang
parent 97bf79917c
commit 7659cae3ba
3 changed files with 11 additions and 2 deletions
+1 -1
View File
@@ -213,7 +213,7 @@ Policy Gradients
Policy gradients architecture (same as A2C)
Tuned examples: `CartPole-v0 <https://github.com/ray-project/ray/blob/master/rllib/tuned_examples/regression_tests/cartpole-pg.yaml>`__
Tuned examples: `CartPole-v0 <https://github.com/ray-project/ray/blob/master/rllib/tuned_examples/regression_tests/cartpole-pg-tf.yaml>`__
**PG-specific configs** (see also `common configs <rllib-training.html#common-parameters>`__):
@@ -1,4 +1,4 @@
cartpole-pg:
cartpole-pg-tf:
env: CartPole-v0
run: PG
stop:
@@ -0,0 +1,9 @@
cartpole-pg-torch:
env: CartPole-v0
run: PG
stop:
episode_reward_mean: 150
timesteps_total: 100000
config:
num_workers: 0
use_pytorch: true