[RLlib] PyTorch version of ARS (Augmented Random Search). (#8106)

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.
This commit is contained in:
Sven Mika
2020-04-21 09:47:52 +02:00
committed by GitHub
parent d66d12661b
commit d15609ba2a
16 changed files with 199 additions and 292 deletions
+2 -2
View File
@@ -12,7 +12,7 @@ Feature Compatibility Matrix
Algorithm Frameworks Discrete Actions Continuous Actions Multi-Agent Model Support
=================== ========== ======================= ================== =========== =====================
`A2C, A3C`_ tf + torch **Yes** `+parametric`_ **Yes** **Yes** `+RNN`_, `+autoreg`_
`ARS`_ tf **Yes** **Yes** No
`ARS`_ tf + torch **Yes** **Yes** No
`ES`_ tf + torch **Yes** **Yes** No
`DDPG`_, `TD3`_ tf + torch No **Yes** **Yes**
`APEX-DDPG`_ tf No **Yes** **Yes**
@@ -405,7 +405,7 @@ Derivative-free
Augmented Random Search (ARS)
-----------------------------
|tensorflow|
|pytorch| |tensorflow|
`[paper] <https://arxiv.org/abs/1803.07055>`__ `[implementation] <https://github.com/ray-project/ray/blob/master/rllib/agents/ars/ars.py>`__
ARS is a random search method for training linear policies for continuous control problems. Code here is adapted from https://github.com/modestyachts/ARS to integrate with RLlib APIs.
+1 -1
View File
@@ -113,7 +113,7 @@ Algorithms
* Derivative-free
- |tensorflow| :ref:`Augmented Random Search (ARS) <ars>`
- |pytorch| |tensorflow| :ref:`Augmented Random Search (ARS) <ars>`
- |pytorch| |tensorflow| :ref:`Evolution Strategies <es>`