-
Notifications
You must be signed in to change notification settings - Fork 2.1k
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.
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
torch2trt 转换问题 #1742
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
我在使用torch2trt转我的yolox模型的时候,原本可以转,但是我是用repvggblock替换原本的cspblock的时候,torch2trt就会失效,不会出现任何错误而是直接中止转换。就像这样(yolox) R:\YOLOX>python tool/trt.py --onlypth
2023-12-07 20:53:47.909 | INFO | main:main:63 - loaded checkpoint done.
[12/07/2023-20:53:50] [TRT] [I] [MemUsageChange] Init CUDA: CPU +776, GPU +0, now: CPU 13024, GPU 2274 (MiB)
(yolox) R:\YOLOX>。但是使用pth to onnx 再从onnx to engine是可行的,但是不会生成model_trt文件。能否告知一下为什么torch2trt不支持repvggblock。或者能提供一个调用python api的推理脚本吗,方便使用engine文件。十分感谢大佬!
The text was updated successfully, but these errors were encountered: