婷婷综合国产,91蜜桃婷婷狠狠久久综合9色 ,九九九九九精品,国产综合av

主頁 > 知識庫 > Pytorch dataloader在加載最后一個batch時卡死的解決

Pytorch dataloader在加載最后一個batch時卡死的解決

熱門標簽:開封語音外呼系統代理商 天津電話機器人公司 河北防封卡電銷卡 開封自動外呼系統怎么收費 地圖標注線上如何操作 電銷機器人的風險 手機網頁嵌入地圖標注位置 400電話辦理哪種 應電話機器人打電話違法嗎

問題:

自己寫了個dataloader,為了部署方便,用OpenCV的接口進行數據讀取,而沒有用PIL,代碼大致如下:

    def __getitem__(self, idx):
        sample = self.samples[idx]
 
        img = cv2.imread(sample[0])
        img = cv2.resize(img, tuple(self.input_size))
        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        # if not self.val and random.randint(1, 10)  3:
        #     img = self.img_aug(img)
        img = Image.fromarray(img) 
        img = self.transforms(img)        
        ...

結果在訓練過程中,在第1個epoch的最后一個batch時,程序卡死。

解決方案:

可能是因為OpenCV與Pytorch互鎖的問題,關閉OpenCV的多線程,問題解決。

cv2.setNumThreads(0)
cv2.ocl.setUseOpenCL(False)

補充:pytorch 中一個batch的訓練過程

# 一般情況下
optimizer.zero_grad()             # 梯度清零
preds = model(inputs)             # inference,前向傳播求出預測值
loss = criterion(preds, targets)  # 計算loss
loss.backward()                   # 反向傳播求解梯度
optimizer.step()                  # 更新權重,更新網絡權重參數

此外,反向傳播前,如果不進行梯度清零,則可以實現梯度累加,從而一定程度上解決顯存受限的問題。

以上為個人經驗,希望能給大家一個參考,也希望大家多多支持腳本之家。

您可能感興趣的文章:
  • pytorch鎖死在dataloader(訓練時卡死)
  • pytorch Dataset,DataLoader產生自定義的訓練數據案例
  • 解決Pytorch dataloader時報錯每個tensor維度不一樣的問題
  • pytorch中DataLoader()過程中遇到的一些問題
  • Pytorch 如何加速Dataloader提升數據讀取速度
  • pytorch DataLoader的num_workers參數與設置大小詳解
  • pytorch 實現多個Dataloader同時訓練

標簽:江蘇 蘭州 駐馬店 六盤水 常州 成都 宿遷 山東

巨人網絡通訊聲明:本文標題《Pytorch dataloader在加載最后一個batch時卡死的解決》,本文關鍵詞  Pytorch,dataloader,在,加載,;如發現本文內容存在版權問題,煩請提供相關信息告之我們,我們將及時溝通與處理。本站內容系統采集于網絡,涉及言論、版權與本站無關。
  • 相關文章
  • 下面列出與本文章《Pytorch dataloader在加載最后一個batch時卡死的解決》相關的同類信息!
  • 本頁收集關于Pytorch dataloader在加載最后一個batch時卡死的解決的相關信息資訊供網民參考!
  • 推薦文章
    主站蜘蛛池模板: 绥化市| 将乐县| 隆尧县| 广州市| 闵行区| 登封市| 电白县| 务川| 庄河市| 克什克腾旗| 绥滨县| 广元市| 淳化县| 城固县| 江山市| 张家港市| 奉新县| 隆安县| 榆社县| 松阳县| 郸城县| 西乌| 靖宇县| 临高县| 莱西市| 蒲江县| 莱芜市| 南岸区| 安达市| 英山县| 竹山县| 天台县| 十堰市| 鸡东县| 福州市| 江津市| 盐亭县| 河北区| 广德县| 会昌县| 赤水市|