Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/goktug97/nes-torch
Minimal PyTorch Library for Natural Evolution Strategies
https://github.com/goktug97/nes-torch
evolution-strategies evolution-strategy evolutionary-algorithms natural-evolution-strategies policy-gradient pytorch reinforcement-learning
Last synced: about 2 months ago
JSON representation
Minimal PyTorch Library for Natural Evolution Strategies
- Host: GitHub
- URL: https://github.com/goktug97/nes-torch
- Owner: goktug97
- License: mit
- Created: 2021-04-16T13:15:40.000Z (over 3 years ago)
- Default Branch: master
- Last Pushed: 2021-09-29T19:40:58.000Z (about 3 years ago)
- Last Synced: 2024-11-08T02:42:18.213Z (about 2 months ago)
- Topics: evolution-strategies, evolution-strategy, evolutionary-algorithms, natural-evolution-strategies, policy-gradient, pytorch, reinforcement-learning
- Language: Python
- Homepage:
- Size: 755 KB
- Stars: 17
- Watchers: 3
- Forks: 3
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
Minimal PyTorch Library for Natural Evolution Strategies
========================================================
# Requirements
```
numpy
torch
```
## Optional
```
gym # For gym examples
mpi4py # For parallelization
```
To install `mpi4py` you need `MPI` installed in the system.
Check: https://mpi4py.readthedocs.io/en/stable/install.html
A Dockerfile is provided for convenience.
# Installation
```bash
pip install nes-torch --user
```
# Documentation
https://nestorch.readthedocs.io/
# Usage
See https://github.com/goktug97/nes-torch/blob/master/examples
Check documentation or https://github.com/goktug97/nes-torch/blob/master/nes/config.py for parameters.
You can run the example with
```bash
PYTHONPATH="$(pwd):$PYTHONPATH" python examples/example.py
```
or in parallel for faster training.
```bash
PYTHONPATH="$(pwd):$PYTHONPATH" mpirun --np 2 python examples/example.py
```