search

mit-han-lab / efficientvit

EfficientViT is a new family of vision models for efficient high-resolution vision.
Apache License 2.0
1.79k stars 164 forks source link

Tensorflow Lite models #3

Closed gaetanbahl closed 1 year ago

gaetanbahl commented 1 year ago

Hello, thanks for this work.

Could you share the Tensorflow lite models that were used for benchmarking? Or even better, could you also share the code to convert from PyTorch to TFLite? I would like to try quantizing the models.

Thanks in advance.

han-cai commented 1 year ago

Hi,

I have added the code for converting Pytorch models to TFLite models: https://github.com/mit-han-lab/efficientvit/blob/master/export_tflite.py

I will close this issue for now. Feel free to contact me if you have any further questions.

Best regards, Han