Rspress

# #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_model | ModelProto | 必选，无默认值 | 输入的onnx模型。
march | str | 必选，无默认值 | 计算平台。
input_dict | Optional[Dict[str, Any]] | None | 根据指定的参数，修改转换后的模型输入。注意：调用者需要确保对模型的输入修改是合法的。 修改指定输入的input_shape。 `input_dict = {'input_name0': {'input_shape':[x,x,x,x]}}` 修改指定输入的batch_size。 `input_dict = {'input_name0': {'input_batch': 4}}`

页面目录