pytorch转tensorrt_pytorch是啥

全栈程序员站长

发布于 2022-09-27 11:39:12

2640

发布于 2022-09-27 11:39:12

文章被收录于专栏：全栈程序员必看

报错

涉及到梯度传播的tensor变量不能够直接numpy()转numpy，需要使用：

var.detach().numpy()

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/183816.html原文链接：https://javaforall.cn

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度