search

FITI-HCITA / fitipower_sic_2024

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

我的模型是pytorch模型該怎麼辦? #1

Open ATMRORO opened 2 months ago

ATMRORO commented 2 months ago

pytorch model how to convert ?

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

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