不计算梯度 使用with torch.no_grad():

请问这个在YOLOV5源码的哪个文件下的代码里面加呀,不计算梯度
使用with torch.no_grad():

img

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^