English

hmct.api.check_model

Interface Description

The quick check function for the conversion process of the model is provided by HMCT, where the input onnx model is checked with some random quantization parameters to see if the model conversion and the process is successful.

Interface Form

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

Return Value

Generate an onnx model with stochastic parameter quantization.

Parameter

The value ranges and definitions of the same parameters in check_model as in build_model are identical.

Parameter	Type	Default Value	Description
onnx_model	ModelProto	Required, no default value	The input onnx model.
march	str	Required, no default value	Computing platform.
input_dict	Optional[Dict[str, Any]]	None	Modify the converted model inputs according to the specified parameters. Note: The caller needs to ensure that the modifications to the model inputs are legal. Modify the input_shape of the specified input: `input_dict = {'input_name0': {'input_shape':[x,x,x,x]}}` Modify the batch_size of the specified input: `input_dict = {'input_name0': {'input_batch': 4}}`

On This Page

PTQ Conversion Tools

The hb_compile Tool

PTQ Conversion Steps

PTQ Conversion Samples

FAQ and Common Failure Resolutions

Appendix

Tutorial

Advanced Tutorial

API Reference

QAT

qconfig_setter

Model Export

Horizon Operator

FAQ and Common Failure Resolutions

Model Inference Development

Model Inference API Introduction

Data Type and Data Structure

Function Interface

Model Inference Tool Introduction

The hrt_model_exec Tool Introduction

The hbm_infer Tool Introduction

UCP Common API Introduction

Data Structure

Functional Interface

UCP Performance Analysis Tools

FAQ and Error code

Model Deployment Principle and Process

Model Deployment Practice Guidance Examples

HMCT API Reference

Toolchain Operator Support Constraint List

Operator Support List

Operator BPU Constraint List

Community Quality Articles

hmct.api.check_model

Interface Description

Interface Form

Return Value

Parameter

The hb_compile Tool

QAT

qconfig_setter

Model Export

Horizon Operator

Model Inference API Introduction

Data Type and Data Structure

Function Interface

Model Inference Tool Introduction

The hrt_model_exec Tool Introduction

The hbm_infer Tool Introduction

Data Structure

Functional Interface

Operator Support List

Operator BPU Constraint List

#hmct.api.check_model

#Interface Description

#Interface Form

#Return Value

#Parameter

hmct.api.check_model

Interface Description

Interface Form

Return Value

Parameter