将onnx转换为Tensorrt模型时找不到ScatterND插件

问题描述

能够将pytorch模型转换为onnx并使用pytorch验证onnx输出。工作正常。 但是当我尝试将onnx转换为tensorrt引擎时,它在onnx解析器上失败并返回此错误

[TensorRT]错误:INVALID_ARGUMENT:getPluginCreator找不到插件ScatterND版本1

Onnx版本:1.6.0 Tensorrt版本:7.1.3.4

任何人都可以在这里提供帮助,如何在python中添加ScatterND插件以及如何使用它? 还有其他方法可以解决此问题吗?

谢谢。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)