網(wǎng)站首頁(yè) 編程語(yǔ)言 正文
RuntimeError: a view of a leaf Variable that requires grad is being used in an in-place
作者:別出BUG求求了 更新時(shí)間: 2023-11-17 編程語(yǔ)言yolov5報(bào)錯(cuò):RuntimeError: a view of a leaf Variable that requires grad is being used in an in-place
解決辦法:
在model/yolo.py文件
for mi, s in zip(m.m, m.stride): # from
b = mi.bias.view(m.na, -1) # conv.bias(255) to (3,85)
b[:, 4] += math.log(8 / (640 / s) ** 2) # obj (8 objects per 640 image)
b[:, 5:] += math.log(0.6 / (m.nc - 0.99)) if cf is None else torch.log(cf / cf.sum()) # cls
mi.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
添加with torch.no_grad():如下
for mi, s in zip(m.m, m.stride): # from
b = mi.bias.view(m.na, -1) # conv.bias(255) to (3,85)
with torch.no_grad():
b[:, 4] += math.log(8 / (640 / s) ** 2) # obj (8 objects per 640 image)
b[:, 5:] += math.log(0.6 / (m.nc - 0.99)) if cf is None else torch.log(cf / cf.sum()) # cls
mi.bias = torch.nn.Parameter(b.view(-1), requires_grad=True)
——> 根本原因就是在報(bào)錯(cuò)的語(yǔ)句前添加:
with torch.no_grad():
原文鏈接:https://blog.csdn.net/weixin_39589455/article/details/126808770
- 上一篇:沒(méi)有了
- 下一篇:沒(méi)有了
相關(guān)推薦
- 2023-01-19 React?classnames原理及測(cè)試用例_React
- 2022-09-30 在ubuntu下安裝go開(kāi)發(fā)環(huán)境的全過(guò)程_Golang
- 2022-12-05 Go?reflect?反射原理示例詳解_Golang
- 2022-04-18 Android實(shí)現(xiàn)繪制折線圖APP代碼_Android
- 2022-09-10 Nginx代理Partainer的實(shí)現(xiàn)_nginx
- 2022-06-13 golang下grpc框架的使用編寫示例_Golang
- 2022-08-23 Shell?腳本自動(dòng)輸入密碼的三種方式小結(jié)_linux shell
- 2023-01-28 ajax、axios和fetch之間優(yōu)缺點(diǎn)重點(diǎn)對(duì)比總結(jié)_AJAX相關(guān)
- 欄目分類
-
- 最近更新
-
- window11 系統(tǒng)安裝 yarn
- 超詳細(xì)win安裝深度學(xué)習(xí)環(huán)境2025年最新版(
- Linux 中運(yùn)行的top命令 怎么退出?
- MySQL 中decimal 的用法? 存儲(chǔ)小
- get 、set 、toString 方法的使
- @Resource和 @Autowired注解
- Java基礎(chǔ)操作-- 運(yùn)算符,流程控制 Flo
- 1. Int 和Integer 的區(qū)別,Jav
- spring @retryable不生效的一種
- Spring Security之認(rèn)證信息的處理
- Spring Security之認(rèn)證過(guò)濾器
- Spring Security概述快速入門
- Spring Security之配置體系
- 【SpringBoot】SpringCache
- Spring Security之基于方法配置權(quán)
- redisson分布式鎖中waittime的設(shè)
- maven:解決release錯(cuò)誤:Artif
- restTemplate使用總結(jié)
- Spring Security之安全異常處理
- MybatisPlus優(yōu)雅實(shí)現(xiàn)加密?
- Spring ioc容器與Bean的生命周期。
- 【探索SpringCloud】服務(wù)發(fā)現(xiàn)-Nac
- Spring Security之基于HttpR
- Redis 底層數(shù)據(jù)結(jié)構(gòu)-簡(jiǎn)單動(dòng)態(tài)字符串(SD
- arthas操作spring被代理目標(biāo)對(duì)象命令
- Spring中的單例模式應(yīng)用詳解
- 聊聊消息隊(duì)列,發(fā)送消息的4種方式
- bootspring第三方資源配置管理
- GIT同步修改后的遠(yuǎn)程分支