DanaHan/Yolov5-in-Deepstream-5.0

issues with serializing into engine file

Opened this issue · 1 comments

Hi

i having issue trying to convert a generated wts (based on 3.1) into engine file when executing yolov5. Can anyone advise?

oading weights: yolov5l_3.1.wts
[10/13/2021-14:41:31] [E] [TRT] Parameter check failed at: ../builder/Network.cpp::addScale::494, condition: shift.count > 0 ? (shift.values != nullptr) : (shift.values == nullptr)
yolov5: /home/yolov5/common.hpp:155: nvinfer1::IScaleLayer* addBatchNorm2d(nvinfer1::INetworkDefinition*, std::map<std::__cxx11::basic_string, nvinfer1::Weights>&, nvinfer1::ITensor&, std::string, float): Assertion `scale_1' failed.
Aborted (core dumped)

@lchunleo have you managed to solve this please?