yolov5s v1.0 序列化报错

[hoxcape]

我使用yolov5 v1.0训练了yolov5s模型，然后进行序列化，得到以下信息：

Loading weights: ../yolov5s.wts
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: kernel weights has count 2304 but 32640 was expected
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: count of 2304 weights in kernel, but kernel dimensions (1,1) with 128 input channels, 255 output channels and 1 groups were specified. Expected Weights count is 128 * 1*1 * 255 / 1 = 32640
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: kernel weights has count 2304 but 32640 was expected
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: count of 2304 weights in kernel, but kernel dimensions (1,1) with 128 input channels, 255 output channels and 1 groups were specified. Expected Weights count is 128 * 1*1 * 255 / 1 = 32640
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: kernel weights has count 2304 but 32640 was expected
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: count of 2304 weights in kernel, but kernel dimensions (1,1) with 128 input channels, 255 output channels and 1 groups were specified. Expected Weights count is 128 * 1*1 * 255 / 1 = 32640
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: kernel weights has count 2304 but 32640 was expected
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: count of 2304 weights in kernel, but kernel dimensions (1,1) with 128 input channels, 255 output channels and 1 groups were specified. Expected Weights count is 128 * 1*1 * 255 / 1 = 32640
Building engine, please wait for a while...
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: kernel weights has count 2304 but 32640 was expected
[07/31/2020-11:37:13] [E] [TRT] (Unnamed Layer* 173) [Convolution]: count of 2304 weights in kernel, but kernel dimensions (1,1) with 128 input channels, 255 output channels and 1 groups were specified. Expected Weights count is 128 * 1*1 * 255 / 1 = 32640
[07/31/2020-11:37:13] [E] [TRT] Could not compute dimensions for (Unnamed Layer* 173) [Convolution]_output, because the network is not valid.
[07/31/2020-11:37:13] [E] [TRT] Network validation failed.
Build engine successfully!
yolov5s: yolov5_tensorrt/yolov5s/yolov5s.cpp:170: void APIToModel(unsigned int, nvinfer1::IHostMemory**): Assertion `engine != nullptr' failed.
已放弃 (核心已转储)

请帮我看一下大概是什么问题，谢谢了。