给位大神:我用https://github.com/biubug6/Pytorch_Retinaface这个工程的mobilenet0.25骨干网络训练了自己的pth模型,用torch自带工具转onnx没有问题,但是用TensorRT自带转换工具转成trt时报如下错误:

Assertion failed: scales_input.is_weights()

d45ba333-0280-4654-b5b1-54a36c2830ec-图片.png

社区算法库有个retinaface_mbv2_leaky.onnx可以成功转换成trt,但没有搜到训练代码:(
请问大神能否提供一个可以使用TensorRT加速的retinaface模型训练代码?感谢!