You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
**GenRL is a PyTorch reinforcement learning library centered around reproducible and generalizable algorithm implementations.**
24
+
**GenRL is a PyTorch reinforcement learning library centered around reproducible, generalizable algorithm implementations and improving accessibility in Reinforcement Learning**
24
25
25
26
Reinforcement learning research is moving faster than ever before. In order to keep up with the growing trend and ensure that RL research remains reproducible, GenRL aims to aid faster paper reproduction and benchmarking by providing the following main features:
26
27
27
28
-**PyTorch-first**: Modular, Extensible and Idiomatic Python
29
+
-**Tutorials and Example**: 20+ Tutorials from basic RL to SOTA Deep RL algorithm (with explanations)!
28
30
-**Unified Trainer and Logging class**: code reusability and high-level UI
29
31
-**Ready-made algorithm implementations**: ready-made implementations of popular RL algorithms.
30
32
-**Faster Benchmarking**: automated hyperparameter tuning, environment implementations etc.
31
33
32
34
By integrating these features into GenRL, we aim to eventually support **any new algorithm implementation in less than 100 lines**.
33
35
34
-
**If you're interested in contributing, feel free to go through the issues and open PRs for code, docs, tests etc. In case of any questions, please check out the [Contributing Guidelines](https://github.com/SforAiDl/genrl/wiki/Contributing-Guidelines)**
36
+
**If you're interested in contributing, feel free to go through the issues and open PRs for code, docs, tests etc. In case of any questions, please check out the [Contributing Guidelines](CONTRIBUTING.md)**
35
37
36
38
37
39
## Installation
@@ -55,10 +57,9 @@ To train a Soft Actor-Critic model from scratch on the `Pendulum-v0` gym environ
55
57
```python
56
58
import gym
57
59
58
-
from genrl importSAC, QLearning
59
-
from genrl.classical.common import Trainer
60
-
from genrl.deep.common import OffPolicyTrainer
60
+
from genrl.agents importSAC, QLearning
61
61
from genrl.environments import VectorEnv
62
+
from genrl.trainers import ClassicalTrainer, OffPolicyTrainer
62
63
63
64
env = VectorEnv("Pendulum-v0")
64
65
agent = SAC('mlp', env)
@@ -69,13 +70,30 @@ trainer.train()
69
70
To train a Tabular Dyna-Q model from scratch on the `FrozenLake-v0` gym environment and plot rewards:
0 commit comments