KeyError: 'engine'

[caikw0602]

博主你好：
再用你的模型时候，生成.engine时，发生了错误，提示为KeyError：'engine'，我的模型是pytorch的.pt模型

[cmcamdy]

你好，初步判断可能是模型加载那块有点问题，能否附上详细些的信息，或者可以参考以下步骤：
1.确认模型是否正确加载
2.debug进入torch2trt初始化页面，查看出错位置，查看是否因为某些原因导致engine加载不成功，因为它这个engine是包在TRTMODULE中的，个人猜测如果engine没有加载好可能会发生这个错误
3.其他

[caikw0602]

我试着把tensorrt转换关掉，也有问题，我用的是u版的yolov3训练的模型

[caikw0602]

我打开tensorrt转换后，报错信息是：

[cmcamdy]

你好，本项目不是u版的，是基于https://github.com/eriklindernoren/PyTorch-YOLOv3 的
可以参考二者之间的区别：https://blog.csdn.net/nihate/article/details/103736384
或许会有发现