Я использую предварительно обученную модель Yolov5, я успешно создал yolov5s. Файл движка, но что-то проблема в файле yolov5_trt.py. Ядро сброшено

Я следую всем указанным инструкциям (https://github.com/wang-xinyu/tensorrtx/tree/master/yolov5). Ошибка при запуске файла python yolov5_trt.py.

python: engine.cpp: 1104: bool nvinfer1 :: rt :: Engine :: deserialize (const void *, std :: size_t, nvinfer1 :: IGpuAllocator &, nvinfer1 :: IPluginFactory *): Assertion `size› = bsize && Несоответствие между размер выделенной памяти и ожидаемый размер сериализованного движка. ' не смогли. Прервано (ядро выгружено)

TensorRT-5.1.5 os: Ubuntu 16.04 torch: 1.6.0 + cu101 torchvision: 0.7.0 + cu101 cuda: 10.1 python: 3.6

Спасибо...