nes-torch 0.0.2

Minimal PyTorch Library for Natural Evolution Strategies

Minimal PyTorch Library for Natural Evolution Strategies

Requirements

pipcs
numpy
torch
gym
mpi4py
box2d-py

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

pip install nes-torch --user

Documentation

https://nestorch.readthedocs.io/

Usage

See https://github.com/goktug97/nes-torch/blob/master/example.py

Check https://github.com/goktug97/pipcs to understand the configuration system.

Check documentation or https://github.com/goktug97/nes-torch/blob/master/nes/config.py for parameters.

You can run the example with

python example.py

or in parallel for faster training.

mpirun -np 2 python example.py