Simple converter for deploying Stable-Baselines3 model to TFLite and/or Coral
Introduction I’ve been using Stable-Baselines3to train agents against some custom Gyms, some of which require fairlylarge NNs in order to be effective. I want those agents to eventually be run on a pi or similar, so I needto export all the way to TFLiteand 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 […]
Read more