- Notifications
You must be signed in to change notification settings - Fork1
A collection of Reinforcement Learning implementations with PyTorch
License
hcnoh/rl-collection-pytorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
This repository is a collection of the following reinforcement learning algorithms:
- Policy-Gradient
- Actor-Critic
- Trust Region Policy Optimization
- Generalized Advantage Estimation
- Proximal Policy Optimization
More algorithms will be added on this repository.
In this repository,OpenAI Gym environments such asCartPole-v0,Pendulum-v0, andBipedalWalker-v3 are used. You need to install them before running this repository.
Note: The environment's names could be different depending on the version of OpenAI Gym.
Install Python 3.
Install the Python packages in
requirements.txt. If you are using a virtual environment for Python package management, you can install all python packages needed by using the following bash command:$ pip install -r requirements.txt
Install other packages to run OpenAI Gym environments. These are dependent on the development setting of your machine.
Install PyTorch. The version of PyTorch should be greater or equal than 1.7.0.
Modify
config.jsonas your machine setting.Execute training process by
train.py. An example of usage fortrain.pyare following:$ python train.py --model_name=trpo --env_name=BipedalWalker-v3
The following bash command will help you:
$ python train.py -h
You can run your pre-trained agents by executing
run.py. The usage for runningrun.pyis similar to that oftrain.py. You can also check the help message by the following bash bash command:$ python run.py -h
- The CUDA usage is provided now.
- Modified some errors in GAE and PPO.
- Modified some errors about horizon was corrected.
- Find the errors of the Actor-Critic
- Implement ACER
- Search other environments to running the algorithms
About
A collection of Reinforcement Learning implementations with PyTorch
Topics
Resources
License
Uh oh!
There was an error while loading.Please reload this page.





