907 B

Raw Permalink Blame History

tiny-rl

NOTE: Currently only stable baselines is supported out of the box

The goal of this repository is to make an easy-to-use framework that wraps the QPD algorithm to compress models of RL policies.

Installation

First these libraries are required to install.

sudo apt update
sudo apt install libpng-dev libjpeg-dev zlib1g-dev

Afterwards install the python library.

Important: Make sure u use python version 3.10.*, setup tools version 65.0 and wheel 0.38.0

conda create -n compression python=3.10
conda activate compression
python -m pip install setuptools==65.0 wheel==0.38.4
python -m pip install -e .

Currently supported

A2C
PPO
DQN
QRDQN
SAC

Cheetah env installation if used with conda

conda install -c conda-forge xorg-libx11
conda install -c conda-forge glew
conda install -c conda-forge mesalib
conda install -c menpo glfw3