【機器學習】pytorch中Dataloader的坑

基於pytorch訓練一個模型,數據集size大概七十萬 訓練起來很慢 瘋狂找緣由,後來通過一行行調試,發現是遍歷Dataloader時太慢了 正在尋找解決辦法。。(待更新2020.1.5)python 排坑1 :有大佬說PIL庫比opencv慢,因而把代碼裏PIL庫打開圖片操做改爲了opencvweb img = Image.open(img_name) #省略xxx行 t1 = time.ti
相關文章
相關標籤/搜索