1ee5315ea20d845952e8938a57b7ecd85ed14a15,mmcv/tensorrt/tensorrt_utils.py,,onnx2trt,#,5

Before Change

    if isinstance(onnx_model, str):
        assert parser.parse_from_file(onnx_model), "parse onnx failed."
    else:
        assert parser.parse(
            onnx_model.SerializeToString()), "parse onnx failed."

    // config builder
    builder.max_workspace_size = max_workspace_size

After Change

    if not parser.parse(onnx_model.SerializeToString()):
        error_msgs = ""
        for error in range(parser.num_errors):
            error_msgs += f"{parser.get_error(error)}\n"
        raise RuntimeError(f"parse onnx failed:\n{error_msgs}")

    // config builder
    builder.max_workspace_size = max_workspace_size

Instances