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}}`