Yunfeng's Simple Blog
·
Pytorch转ONNX报错-Cannot insert a Tensor that requires grad as a constant
💡
原文中文，约800字，阅读约需2分钟。
📝
内容提要
在将Pytorch模型转换为ONNX时，遇到“无法将需要梯度的Tensor作为常量”的错误。问题出在Conv2D层的权重设置为requires_grad为True。最终解决方案是将模型所有层的参数设置为requires_grad = False。
🎯
关键要点
- 在将Pytorch模型转换为ONNX时，遇到错误：无法将需要梯度的Tensor作为常量。
- 错误源于Conv2D层的权重设置为requires_grad为True。
- 尝试了多种方法解决问题，但均未成功。
- 最终解决方案是将模型所有层的参数设置为requires_grad = False。
➡️
