Skip to content

Commit 08a6c8a

Browse files
Merge pull request #13 from mohammadzainabbas/dev
Dev
2 parents a8e55c1 + 02a7003 commit 08a6c8a

File tree

9 files changed

+19396
-0
lines changed

9 files changed

+19396
-0
lines changed

β€ŽREADME.mdβ€Ž

Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,4 +26,64 @@ This repository contains the code and resources for a reinforcement learning pro
2626

2727
These instructions will get you a copy of the project up and running on your local machine for development and testing purposes.
2828

29+
#
30+
31+
<a id="prerequisites" />
32+
33+
#### 2.1. βš™οΈ Prerequisites βš™οΈ
34+
35+
- [x] Python 3.7 or higher
36+
- [x] Brax 0.1.0 or higher
37+
- [x] Jax 0.4.1 or higher
38+
- [x] Flax 0.6.3 or higher
39+
- [x] PyTorch 1.13.1 or higher
40+
41+
#
42+
43+
<a id="installation" />
44+
45+
#### 2.2. πŸ”§ Installation πŸ”§
46+
47+
1. Clone the repository
48+
49+
```bash
50+
git clone https://github.com/mohammadzainabbas/Reinforcement-Learning-CS.git
51+
cd Reinforcement-Learning-CS/
52+
```
53+
54+
2. Install the requirements
55+
56+
```bash
57+
pip install -r requirements.txt
58+
```
59+
60+
#
61+
62+
<a id="running-demos" />
63+
64+
#### 2.3. πŸƒβ€β™‚οΈ Running Demo(s) πŸƒβ€β™‚οΈ
65+
66+
- [x] [Grasp: Pick-and-Place with a robotic hand](https://colab.research.google.com/github/mohammadzainabbas/Reinforcement-Learning-CS/blob/main/notebooks/demo.ipynb)
67+
- [x] [Step-by-step training with PPO](https://colab.research.google.com/github/mohammadzainabbas/Reinforcement-Learning-CS/blob/main/notebooks/demo_ppo_train.ipynb)
68+
69+
Or you can run the `train_ppo.py` file locally by following the steps below:
70+
71+
```bash
72+
python src/train_ppo.py
73+
```
74+
75+
You will get the following output files:
76+
77+
1. `ppo_training.png` - Training progress plot
78+
2. `result_with_ppo.html` - Simulation of the trained agent (in HTML format)
79+
3. `ppo_params` - Trained parameters of the agent
80+
81+
#
82+
83+
<a id="results" />
84+
85+
#### 2.4. πŸŽ‰ Results πŸŽ‰
86+
87+
The following plot shows the training progress of the agent:
2988

89+
![Training progress plot](https://github.com/mohammadzainabbas/Reinforcement-Learning-CS/blob/main/results/output.png?raw=true)

β€Žresults/data/final_600M.jsonβ€Ž

Whitespace-only changes.

0 commit comments

Comments
Β (0)