NX tensorrt7.1 UNKNOWN:Bias weights are not set yet

The tensorrt code is ok on pc. But is not work on jetson nx, the out message is

UNKNOWN:Bias weights are not set yet. Bias weights can be set using setINput(2,bias_tensor) API call.

Hi,

Could you share which TensorRT version are you using for desktop?
And would you mind to share a simple reproducible sample so we can check it directly?

Thanks.

pc:windows tensorrt7,1 cuda10.0 cudnn7.5
nx: JetPack 4,4

and the sample code is here : https://github.com/enazoe/yolo-tensorrt

hi AastaLLL:
when I comment the line:

set(CUDA_NVCC_FLAGS${CUDA_NVCC_FLAGS}; -gencode arch=compute_53,code=sm_53)

at https://github.com/enazoe/yolo-tensorrt/blob/522756a308538c90e757918618b9babdad4820af/CMakeLists.txt#L27
and the code could work.But the program rather output the UNKNOW messange.

And I want to know the compute capability of the jetson nx board?

Hi,

Please use the sm=72 for XavierNX.

set(CUDA_NVCC_FLAGS${CUDA_NVCC_FLAGS}; -gencode arch=compute_72,code=sm_72)

Thanks.