Kernel weights has count 2304 but 32640 was expected

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

Description

Try to build engine in Jetson TX2

I am getting this error

[02/19/2022-11:33:11] [E] [TRT] 3: (Unnamed Layer* 214) [Convolution]:kernel weights has count 2304 but 32640 was expected
[02/19/2022-11:33:11] [E] [TRT] 4: (Unnamed Layer* 214) [Convolution]: count of 2304 weights in kernel, but kernel dimensions (1,1) with 128 input channels, 255 output channels a nd 1 groups were specified. Expected Weights count is 128 * 11 * 255 / 1 = 32640
[02/19/2022-11:33:11] [E] [TRT] 4: [convolutionNode.cpp::computeOutputExtents::28] Error Code 4: Internal Error ((Unnamed Layer 214) [Convolution]: number of kernel weights does not match tensor dimensions)

Environment

TensorRT Version: 8.0
GPU Type: Jetson TX2
Nvidia Driver Version:
CUDA Version: 10.2
CUDNN Version:
Operating System + Version: Ubuntu 18
Python Version (if applicable): Python 3.6
TensorFlow Version (if applicable):
PyTorch Version (if applicable): 1.10
Baremetal or Container (if container which image + tag):

Relevant Files

This is weight file I generated

Steps To Reproduce

toolkits/deploy

Doing deployment part
I made it to build but
I am getting above mentioned error while creating engine

Please help me

Thanks

2

Hi,
This looks like a Jetson issue. Please refer to the below samlples in case useful.

For any further assistance, we recommend you to raise it to the respective platform from the below link

Thanks!