-
Notifications
You must be signed in to change notification settings - Fork 664
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
repvggblock in torch2trt #907
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I encountered an issue when using torch2trt to convert my YOLOX model. It worked fine originally, but when I replaced the original CSPBlock with RepVGGBlock, torch2trt failed to convert the model. There were no error messages, but the conversion process was terminated prematurely. Here is the output:
(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>
I suspect that this issue might be due to some operations in RepVGGBlock not being supported by torch2trt. Could you please provide some guidance on how to resolve this issue? Thank you.
The text was updated successfully, but these errors were encountered: