前往
大廳
主題

10/27

DVB-S2巴巴大俠 | 2022-10-28 03:31:18 | 巴幣 0 | 人氣 207

今天嘗試把做好的Dataset送進yolo訓練,弄到28號2點多才總算能用GPU開始訓練= =,選用的是yolov5。

今天學到(出問題)的點在:

1. labeling的時候使用類別錯亂,導致出錯

解決: 把label.txt檔內標記進行更正,或是重新標記(第一次執行出錯時會產生一個catch檔,要到labels資料夾內刪除,否則下次執行還是會出錯)

2.安裝Pytorch,conda內的cuda抓不到gpu

解決: 裝1.12版本的pytorch,裝1.13的他直接抓不到gpu

3.執行的時候會出現
OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading “D:Anacondaenvspytorch-1.4libsite-package

解決: gpu內存不足,調低workers參數,減少線程或是調低batch_size,我看我的內存仍足,可能是虛擬內存問題
https://blog.csdn.net/Vertira/article/details/127333303 有空再看這篇文章調整虛擬內存,目前用降低workers參數解決。

打到現在3:28了 兩點才開始跑,次數我調500目前123,明天起來看看會是什麼樣子。不過明天是早九,估計我要累死了。


創作回應

更多創作