admin 管理员组

文章数量: 1184232

一个pytorch module导出onnx时提示错误：

RuntimeError: Cannot insert a Tensor that requires grad as a constant. Consider making it a parameter or input, or detaching the gradient
 

我遇到的这个现象比较特殊：

首先是一个常规的python class类里面包含了一个torch nn.Module对象成员，单独导出这个nn.Module为onnx能够成功。

这个python class的成员函数enhance调用了nn.Module计算，然后使用一个torch nn.Module作为wrapper的forward包装该enhance函数导出，即提示上面的错误。

解决方法：

把这个常规的python class改成torch nn.Module的子类，该错误消失。

本文标签： tensor requires RuntimeError insert making

版权声明：本文标题：RuntimeError: Cannot insert a Tensor that requires grad as a constant. Consider making it a paramete 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1754464692a3005276.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。