DBNet.pytorch
DBNet.pytorch copied to clipboard
mobilenetv3 转onnx遇到个错
请问转onnx模型时遇到: Type Error: Type parameter (T) bound to different types (tensor(float) and tensor(int64) in node (Add_4) 大神们遇到了吗,咋解决呀
我转的时候,没有出现该错误但是出现了其他错误,不过已经解决了。
请问转onnx模型时遇到: Type Error: Type parameter (T) bound to different types (tensor(float) and tensor(int64) in node (Add_4) 大神们遇到了吗,咋解决呀
你可以帖下你的pytorch版本,当然,最好直接升级到1.6,这样错误会少很多。
我转的时候,没有出现该错误但是出现了其他错误,不过已经解决了。
您好,请教一下,您是怎么转成ONNX呢?可以贴一些代码吗?蟹蟹~
我转的时候,没有出现该错误但是出现了其他错误,不过已经解决了。
您好,请教一下,您是怎么转成ONNX呢?可以贴一些代码吗?蟹蟹~
你好,请问你解决了吗?我也遇到这个问题
UserWarning: ONNX export failed on threshold because non-zero threshold not supported warnings.warn("ONNX export failed on " + op + " because " + msg + " not supported")他是说一个算子不支持,但是这个算子应该是后处理的分割算子,跟backbone没关系,我这个能转 resnet18的就应该能转 mobilev3的,还是 mobilev3的backbone里有这个算子了?
参考这个:https://github.com/WenmuZhou/DBNet.pytorch/issues/98#issuecomment-901577100 可以成功转换