把onnx模型转TensorRT模型出错

错误一：WARNING: onnx2trt_utils.cpp:366: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.

解决方法

1、安装onnx-simplifier

pip install onnx-simplifier

2、将转化的onnx模型转化为更简单的onnx模型

 python -m onnxsim model_lenet_300.onnx model_lenet_onnxsim_300.onnx