pytorch-ONNX-TensorRT示例相关问题

Question

FelixFu520 opened this issue a year ago · 4 comments

我在windows上跑cookbook中的04-BuildEngineByONNXParser中pyTorch-ONNX-TensorRT示例时，遇到以下错误，

1, createCalibrationAndInferenceData.py中生成的dataDictionary["calibrationData"] = calibrationData 是np.float64的，这个会影响结果么

2，执行main.cpp 打开int8时，会在C++文件夹中calibrator.cpp中报错，报错内容如下

原因时bufferSize是0

Answer 1 · 2023-07-14T01:43:43.000Z

还有个问题，pData未分配内存，当退出构造函数时，会释放，导致结果不对

Answer 2 · 2023-07-25T15:03:43.000Z

遇到了相同的问题，好像是除buffer_size的位置不对，需要往下移。

Answer 3 · 2023-07-25T15:05:49.000Z

还有个问题，pData未分配内存，当退出构造函数时，会释放，导致结果不对

请问这个pData该怎么分配内存。

Answer 4 · 2023-07-26T02:06:59.000Z

看到pr了，感谢