使用jetson nano Ubuntu18.04 部署yolov5时,生成yolov5s.engine文件报错

Description

使用jetson nano Ubuntu18.04 部署yolov5时,生成yolov5s.engine文件报错

Environment

Ubuntu Version: 18.04
Python Version (if applicable): 3.6
PyTorch Version (if applicable): 1.8
YoLov5 Version: V5

Relevant Files

sudo ./yolov5 -s yolov5s.wts yolov5s.engine s
Loading weights: yolov5s.wts
[11/10/2021-20:28:57] [E] [TRT] Parameter check failed at: …/builder/Network.cpp::addScale::482, condition: shift.count > 0 ? (shift.values != nullptr) : (shift.values == nullptr)
yolov5: /home/nano/tensorrtx/yolov5/common.hpp:155: nvinfer1::IScaleLayer* addBatchNorm2d(nvinfer1::INetworkDefinition*, std::map<std::__cxx11::basic_string, nvinfer1::Weights>&, nvinfer1::ITensor&, std::__cxx11::string, float): Assertion `scale_1’ failed.
Aborted

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!