请问这个在YOLOV5源码的哪个文件下的代码里面加呀,不计算梯度使用with torch.no_grad():
torch.no_grad() 是一个上下文管理器,被该语句 wrap 起来的部分将不会track 梯度。
这样可以执行计算,但该计算不会在反向传播中被记录