一般运行 python 脚本遇到这种问题都是给的参数不是严格匹配函数需要的输入的,这种需要具体情况具体分析,但是主要解决就是好好检查自己的代码,函数输入哪里会存在不匹配的地方。
以我下面的代码为例,注册 Pytorch plugin 算子时:
//修改前
def symbolic_plugin(g, arg0, arg1, arg2):
return g.op('plugin', g, arg0, arg1, arg2)
这个可能之前某些情况下确实可以正常运行,但是随着代码更新或者接口调用不同就会遇到上面的问题,主要是 g.op
函数的参数多了一个 g
:
//修改后
def symbolic_plugin(g, arg0, arg1, arg2):
return g.op('plugin', arg0, arg1, arg2)
就可以正常运行了