这是一个Pytorch1.1版本pytorch模型转onnx的bug
参考:https://blog.youkuaiyun.com/Cxiazaiyu/article/details/91129657
解决方法:
1. pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.0.1
2. pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torchvision==0.2.1
本文解决了一个在Pytorch1.1版本中将模型转换为ONNX格式时出现的bug,提供了详细的步骤来安装特定版本的torch和torchvision库,以确保转换过程顺利进行。
1291

被折叠的 条评论
为什么被折叠?



