Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

pytorch-ONNX-TensorRT示例相关问题 #79

Open
FelixFu520 opened this issue Jul 13, 2023 · 4 comments
Open

pytorch-ONNX-TensorRT示例相关问题 #79

FelixFu520 opened this issue Jul 13, 2023 · 4 comments

Comments

@FelixFu520
Copy link

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

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

2，执行main.cpp 打开int8时，会在C++文件夹中calibrator.cpp中报错，报错内容如下
image
原因时bufferSize是0
@FelixFu520
Copy link
Author

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

@FelixFu520 FelixFu520 mentioned this issue Jul 14, 2023
Open
@xuyixuan1999
Copy link

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

@xuyixuan1999
Copy link

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

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

@xuyixuan1999
Copy link

看到pr了，感谢

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@FelixFu520 @xuyixuan1999