Open lchunleo opened 2 years ago
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?
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)