thomasave
/
QPD
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
The code for reproducing the following papers: - https://hdl.handle.net/10067/2015640151162165141 - https://doi.org/10.3390/s24154876
1 Commit 1 Branch 0 Tags 52 KiB
Python 100%
Go to file
Thomas Avé 862c55e03c Initial public commit 2024-11-23 21:35:13 +01:00
examples Initial public commit 2024-11-23 21:35:13 +01:00
src/qpd Initial public commit 2024-11-23 21:35:13 +01:00
README.md Initial public commit 2024-11-23 21:35:13 +01:00
pyproject.toml Initial public commit 2024-11-23 21:35:13 +01:00

README.md

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