Pytorch转ONNX报错-Cannot insert a Tensor that requires grad as a constant

📝

内容提要

在将Pytorch模型转换为ONNX时，遇到“无法将需要梯度的Tensor作为常量”的错误。问题出在Conv2D层的权重设置为requires_grad为True。最终解决方案是将模型所有层的参数设置为requires_grad = False。

🎯

关键要点

• 在将Pytorch模型转换为ONNX时，遇到错误：无法将需要梯度的Tensor作为常量。

• 错误源于Conv2D层的权重设置为requires_grad为True。

• 尝试了多种方法解决问题，但均未成功。

• 最终解决方案是将模型所有层的参数设置为requires_grad = False。

🏷️

标签

Conv2D ONNX Pytorch requires_grad tensor 模型转换