特牛网址导航

踩坑记录 | pytorch转ONNX时遇到的问题 tinynet smpl_onnx where算子-CSDN博客

网友收藏
文章浏览阅读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算子