FITI-HCITA / fitipower_sic_2024

Fitipower 2024 AI Sustainable Innovation Competition
1 stars 1 forks source link

我的模型是FP32模型該怎麼辦? #2

Open ATMRORO opened 2 months ago

ATMRORO commented 2 months ago

my model is FP32 data type. How can I convert it to INT8 for VA8801?

ATMRORO commented 2 months ago

if you have a pytorch model, the converting steps are:

  1. use pytorch onnx module . try to convert yours pt model file to onnx fp32 model.
  2. Try to find out 3-party tool to support. like onnx2tflite convert it to tensorflow savemodel file.
  3. you need to prepare calbration data and use tensorflow tflite to convert savemodel file to tflite INT8

if your model is tensorflow or keras model, please follow google tensorflow office web to do tflite int8 model convert or follow upper step3.

By the way, some opensource model had support this convert. like yolov5