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

Question

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

Answer 1

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

• 这有个类似的问题, 你可以参考下: https://ask.csdn.net/questions/7495138
• 这篇博客也不错, 你可以看下2021SC@SDUSC山东大学软件学院软件工程应用与实践--YOLOV5代码分析（十五）torch_utils.py
• 除此之外, 这篇博客: Pytorch(一) —— 相关库和函数中的 torch.no_grad() 部分也许能够解决你的问题, 你可以仔细阅读以下内容或者直接跳转源博客中阅读:
  

  torch.no_grad() 是一个上下文管理器，被该语句 wrap 起来的部分将不会track 梯度。

  这样可以执行计算，但该计算不会在反向传播中被记录

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