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

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

💡 原文中文,约800字,阅读约需2分钟。
📝

内容提要

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

🏷️

标签

➡️

继续阅读