/gap_quantization

Set of utilities to quantize and export PyTorch models for GreenWaves GAP8 chip

Primary LanguagePython

Build Status

Intro

Placeholder for GAP8 export and quantization module for PyTorch

Install

    $ sudo pip3 install git+https://github.com/xperience-ai/gap_quantization.git

Installation for development

  1. Clone the repository from github.
  2. Install the development environment:
# create virtual environment using virtualenv (in this example)/pyvenv/conda/etc.
$ virtualenv -p /usr/bin/python3 ./venv
$ . venv/bin/activate
$ pip3 install -r requirements-dev.txt
  1. Instal pre-commit hooks:
$ pre-commit install

After that, all you commit swill be checked with a set of linting tools (some of them are even able to fix the issues!) To run these checks independently from the commit process, just type:

pre-commit run --all-files