快速入门

本章节以压缩特性中的量化功能为例,并借助sample中的网络模型,为您介绍如何使用命令行方式压缩一个模型。

AMCT支持命令行方式和Python API接口方式量化原始网络模型,命令行方式压缩方式,目前仅支持如下ONNX、TensorFlow、Caffe框架:

其他未支持的框架和特性,只能使用Python API方式进行量化。命令行方式相比Python API接口方式有以下优点:

表1 量化方式比较

命令行方式

Python API接口方式

量化准备动作简单,只需准备模型和模型匹配的数据集即可。

需要了解Python语法和量化流程。

量化过程简单,只涉及参数选择,无需对量化脚本进行适配。

需要适配修改量化脚本。

  • 当前仅支持训练后量化中的均匀量化、QAT模型适配CANN模型两个功能。
  • 目前仅支持AMCT(ONNX)、AMCT(TensorFlow)、AMCT(Caffe)。
  • 支持量化的所有功能。
  • 支持所有框架。

压缩ONNX网络模型

压缩开源框架的TensorFlow网络模型

压缩开源框架的Caffe网络模型