maggiez0138 / yolov5_quant_sample

This is 8-bit quantization sample for yolov5. Both PTQ, QAT and Partial Quantization have been implemented, and present the results based on yolov5s.
GNU General Public License v3.0
95 stars 26 forks source link

生成int8.engine报错 #10

Open zmtttt opened 4 months ago

zmtttt commented 4 months ago

engine = builder.build_engine(network, config) [05/07/2024-17:13:08] [TRT] [E] 10: Could not find any implementation for node /model.1/conv/Conv + PWN(PWN(/model.1/act/Sigmoid), /model.1/act/Mul). [05/07/2024-17:13:08] [TRT] [E] 10: [optimizer.cpp::computeCosts::3869] Error Code 10: Internal Error (Could not find any implementation for node /model.1/conv/Conv + PWN(PWN(/model.1/act/Sigmoid), /model.1/act/Mul).) 我的onnx版本为1.16.0;生成的onnx模型的opset为17