RuntimeError: Can't call numpy() on Variable that requires grad. Use var.detach().numpy() instead.

最新推荐文章于 2024-06-27 14:02:08 发布

不悔梦归处sonia

最新推荐文章于 2024-06-27 14:02:08 发布

阅读量1.2w

点赞数 4

本文链接：https://blog.csdn.net/qq_37754309/article/details/90177761

版权

    在运行自己修改后的pytorch程序时，遇到错误RuntimeError: Can't call numpy() on Variable that requires grad. Use var.detach().numpy() instead.
    定位到出错语句为：classes, scores, boxes = [x[:args.top_k].cpu().numpy() for x in t[:3]]
    按照错误提示，将代码修改为：classes, scores, boxes = [x[:args.top_k].cpu().detach().numpy() for x in t[:3]]
    保存后重新运行，不再报错成功运行。