地瓜官网
地瓜开发者社区
地瓜官网
地瓜开发者社区

hmct.api.check_model

接口说明

HMCT提供对模型的转换流程的快速检查功能，对输入的onnx模型，用一些随机的量化参数检查模型的转换和过程是否成功。

接口形式

def check_model( onnx_model: ModelProto, march: str, input_dict: Optional[Dict[str, Any]] = None, ) -> ModelProto

返回值说明

生成一个随机参数量化的onnx模型。

参数说明

check_model中与build_model相同参数的取值范围和定义完全相同。

参数名类型默认值详细说明
onnx_modelModelProto必选，无默认值输入的onnx模型。
marchstr必选，无默认值计算平台。
input_dictOptional[Dict[str, Any]]None

根据指定的参数，修改转换后的模型输入。注意：调用者需要确保对模型的输入修改是合法的。

  1. 修改指定输入的input_shape。
    2. input_dict = {'input_name0': {'input_shape':[x,x,x,x]}}
  2. 修改指定输入的batch_size。
    3. input_dict = {'input_name0': {'input_batch': 4}}