Output 0 of SelectBackward0 is a view and is being modified inplace

AI算法网奇

于 2023-05-26 21:48:58 发布

阅读量922

点赞数

分类专栏： python基础文章标签： python 开发语言

本文链接：https://blog.csdn.net/jacke121/article/details/130894482

版权

python基础专栏收录该内容

465 篇文章 56 订阅

订阅专栏

在运行Yolov5的detect_3d_cls.py脚本时遇到了一个运行时错误，错误信息指出Output0ofSelectBackward0是一个视图并正在原地被修改。问题出在对宽度和高度的条件检查中尝试进行原地修改。解决方案是在相关代码行前添加@torch.no_grad()装饰器，以避免在梯度计算过程中修改视图。

摘要由CSDN通过智能技术生成

yolov5运行报错，

Output 0 of SelectBackward0 is a view and is being modified inplace

File "D:\project\detect\yolov5_wx_55681\det_check\detect_3d_cls.py", line 110, in run
pred = non_max_suppression(pred, conf_thres, iou_thres, opt.classes, opt.agnostic_nms, max_det=max_det)
File "D:\project\detect\yolov5_wx_55681\utils\general.py", line 728, in non_max_suppression
x[((x[..., 2:4] < min_wh) | (x[..., 2:4] > max_wh)).any(1), 4] = 0 # width-height
RuntimeError: Output 0 of SelectBackward0 is a view and is being modified inplace. This view is the output of a function that returns multiple views. Such functions do not allow the output views to be modified inplace. You should replace the inplace operation by an out-of-place one.

解决方法：需要加上@torch.no_grad()