2,851
社区成员




你好,我想知道我是否可以使用 AIMET 工具对我的 PyTorch 模型进行量化并将其转换为 TFLite 模型。我有一个使用 PyTorch 训练的模型,想将其作为 TFLite 模型部署在设备上。我可以使用 AIMET 进行量化吗?
有两种方法可以做到这一点:
PyTorch 模型→AIMET(PyTorch 量化)→ONNX + 编码文件→TensorFlow→TFLite;
PyTorch 模型→TensorFlow→AIMET(TensorFlow 量化)→常规 TF 元数据 / 检查点模型→TFLite。
对于方法 1,你们是否提供将 ONNX 模型和编码文件转换为 TensorFlow 模型的 API?并且是否可以使用方法 2 来进行这个过程呢?
非常感谢。