FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎
GNU General Public License v2.0
1.25k stars 195 forks source link

onnx转换trt报错 #112

Open Yi-hash1 opened 3 months ago

Yi-hash1 commented 3 months ago

[04/10/2024-16:11:31] [W] [TRT] 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. [04/10/2024-16:11:32] [I] Finish parsing network model [04/10/2024-16:11:32] [E] Static model does not take explicit shapes since the shape of inference tensors will be determined by the model itself [04/10/2024-16:11:32] [E] Network And Config setup failed [04/10/2024-16:11:32] [E] Building engine failed [04/10/2024-16:11:32] [E] Failed to create engine from model. [04/10/2024-16:11:32] [E] Engine set up failed

FeiYull commented 3 months ago

@Yi-hash1 导出的onnx是静态输入,可参考仓库中导出onnx的指令