Simple converter for deploying Stable-Baselines3 model to TFLite and/or Coral

Introduction

I’ve been using Stable-Baselines3
to train agents against some custom Gyms, some of which require fairly
large NNs in order to be effective.

I want those agents to eventually be run on a pi or similar, so I need
to export all the way to TFLite
and ideally a Coral.

How to use

Setup

You will need to have configured the Coral system-wide stuff.

Build a venv:

python3 -m venv venv
source venv/bin/activate
python3 -m pip install -r requirements.txt

Running

This comes with enough

 

 

 

To finish reading, please visit source site