/pokesim

Primary LanguagePython

Pokesim

Overview

Pokesim provides:

  • A UNIX server written in Node JS wrapped around the sim and client packages from pkmn
  • A reinforcement learning framework for interacting with this server. Currently supports R-NaD from deepmind, PPO and SAC.

Installation

Run in terminal.

pip install -r requirements
sh ./scripts/download.sh
npm install

Optionally, run this command for installing JAX gpu

pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

Running

In one terminal, run the below command to start the UNIX server

make start

In one another, run the below command to begin training.

python main.py

Configuration

Inside ./config.yml you can change various things about how the algorithm is configured.