issues with serializing into engine file
Opened this issue · 1 comments
lchunleo commented
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)
Emirismail commented
@lchunleo have you managed to solve this please?