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.

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

kernel weights has count 9216 but 87040 was expected #25

Open
mate-hegedus opened this issue Mar 9, 2022 · 0 comments
Open

kernel weights has count 9216 but 87040 was expected #25

mate-hegedus opened this issue Mar 9, 2022 · 0 comments

Comments

@mate-hegedus
Copy link

Please help me out with this one:
I try to convert sample yolov4-p6.pt.. after generating wts i receive this:

"root@3bc22b0a0ace:/yolo/Scaled-YOLOv4-TensorRT/yolov4-p6-tensorrt/build# ./yolov4-p6 -s 1.0 1.0
Loading weights: ../yolov4-p6.wts
Building engine, please wait for a while...
[03/09/2022-19:00:09] [E] [TRT] (Unnamed Layer* 665) [Convolution]: kernel weights has count 9216 but 87040 was expected
[03/09/2022-19:00:09] [E] [TRT] (Unnamed Layer* 665) [Convolution]: count of 9216 weights in kernel, but kernel dimensions (1,1) with 256 input channels, 340 output channels and 1 groups were specified. Expected Weights count is 256 * 11 * 340 / 1 = 87040
[03/09/2022-19:00:09] [E] [TRT] Could not compute dimensions for (Unnamed Layer
665) [Convolution]_output, because the network is not valid.
[03/09/2022-19:00:09] [E] [TRT] Network validation failed.
Build engine successfully!
yolov4-p6: /yolo/Scaled-YOLOv4-TensorRT/yolov4-p6-tensorrt/yolov4-p6.cpp:143: void APIToModel(unsigned int, nvinfer1::IHostMemory**, float&, float&): Assertion `engine != nullptr' failed.
Aborted (core dumped)"

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant