README.md 3.3 KB
Newer Older
1 2 3 4
<p align="center">
<img src=".github/PARL-logo.png" alt="PARL" width="500"/>
</p>

H
Hongsheng Zeng 已提交
5 6
> PARL is a flexible and high-efficient reinforcement learning framework based on [PaddlePaddle](https://github.com/PaddlePaddle/Paddle).

7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
# Features
**Reproducible**. We provide algorithms that stably reproduce the result of many influential reinforcement learning algorithms

**Large Scale**. Ability to support high performance parallelization of training with thousands of CPUs and multi-GPUs 

**Reusable**.  Algorithms provided in repository could be directly adapted to a new task by defining a forward network and training mechanism will be built automatically.

**Extensible**. Build new algorithms quickly by inheriting the abstract class in the framework.


# Abstractions
<img src=".github/abstractions.png" alt="abstractions" width="400"/>  
PARL aims to build an agent for training algorithms to perform complex tasks.   
The main abstractions introduced by PARL that are used to build an agent recursively are the following:

### Model
`Model` is abstracted to construct the forward network which defines a policy network or critic network given state as input.

### Algorithm
`Algorithm` describes the mechanism to update parameters in `Model` and often contains at least one model.

### Agent
B
Bo Zhou 已提交
29
`Agent` is a data bridge between environment and algorithm. It is responsible for data I/O with outside and describes data preprocessing before feeding data into the training process.
30 31 32 33 34 35

Here is an example of building an agent with DQN algorithm for atari games.
```python
import parl
from parl.algorithms import DQN, DDQN

B
Bo Zhou 已提交
36 37 38 39 40 41 42 43
class AtariModel(parl.Model):
	"""AtariModel
	This class defines the forward part for an algorithm,
	its input is state observed on environment.
	"""
	def __init__(self, img_shape, action_dim):
		# define your layers
		self.cnn1 = layers.conv_2d(num_filters=32, filter_size=5,
44
			 			stride=1, padding=2, act='relu')
B
Bo Zhou 已提交
45 46 47 48 49 50 51 52 53
		...
		self.fc1 = layers.fc(action_dim)
	def value(self, img):
		# define how to estimate the Q value based on the image of atari games.
		img = img / 255.0
		l = self.cnn1(img)
		...
		Q = self.fc1(l)
		return Q
54 55
"""
three steps to build an agent
56
   1.  define a forward model which is critic_model in this example
57 58 59 60 61
   2.  a. to build a DQN algorithm, just pass the critic_model to `DQN`
       b. to build a DDQN algorithm, just replace DQN in following line with DDQN
   3.  define the I/O part in AtariAgent so that it could update the algorithm based on the interactive data
"""

B
Bo Zhou 已提交
62 63
model = AtariModel(img_shape=(32, 32), action_dim=4)
algorithm = DQN(model)
64
agent = AtariAgent(algorithm)
65 66 67 68 69
```

# Install:
### Dependencies
- Python 2.7 or 3.5+. 
H
Hongsheng Zeng 已提交
70
- PaddlePaddle >=1.2.1 (We try to make our repository always compatible with latest version PaddlePaddle)  
71 72 73


```
B
Bo Zhou 已提交
74
pip install parl
75 76 77
```

# Examples
H
Hongsheng Zeng 已提交
78 79
- [QuickStart](examples/QuickStart/)
- [DQN](examples/DQN/)
H
Hongsheng Zeng 已提交
80
- [DDPG](examples/DDPG/)
H
Hongsheng Zeng 已提交
81
- [PPO](examples/PPO/)
82
- [IMPALA](examples/IMPALA/)
H
Hongsheng Zeng 已提交
83
- [Winning Solution for NIPS2018: AI for Prosthetics Challenge](examples/NeurIPS2018-AI-for-Prosthetics-Challenge/)
H
Hongsheng Zeng 已提交
84 85 86 87

<img src=".github/NeurlIPS2018.gif" width = "300" height ="200" alt="NeurlIPS2018"/> <img src=".github/Half-Cheetah.gif" width = "300" height ="200" alt="Half-Cheetah"/> <img src=".github/Breakout.gif" width = "200" height ="200" alt="Breakout"/> 
<br>
<img src=".github/Aircraft.gif"  width = "808" height ="300"  alt="NeurlIPS2018"/>