Rspress

# #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