- 链接地址:https://blog.csdn.net/ZYXxxx666/article/details/122810494
- 链接标题:踩坑记录 | pytorch转ONNX时遇到的问题 tinynet smpl_onnx where算子-CSDN博客
- 所属网站:blog.csdn.net
- 被收藏次数:8148
文章浏览阅读3k次。踩坑记录 | PyTorch转ONNX时遇到的问题1. torch环境问题我在转换tinynet模型为onnx时,会产生如下报错ImportError:cannot import name 'container_abcs'这是由于torch版本不对,改为1.7.1就没有这个问题;但对于其他模型对这个版本又不适用,onnx对torch的版本是较为敏感的。转换成功之后还需要验证onnx模型结果,随机生成输入tensor,观察输出结果是否一致,同样注意torch版本问题。2. 去除不支持算子在观察_onnx where算子
版权声明:本文发布于特牛网址导航 内容均来源于互联网 如有侵权联系删除
标签:onnx where算子