1. 程式人生 > >pytorch 深度學習, CPU預處理時間大於GPU網路處理時間、資料增強

pytorch 深度學習, CPU預處理時間大於GPU網路處理時間、資料增強

背景:想知道訓練網路時,時間花在了哪.


做了一個性能分析,關於訓練網路時的時間消耗問題。
1.電腦: i7-6700,1070 ti
2.tiny-YOLO_v1(9層) ,
3.預處理包括隨機平移,縮放,Hue,Saturation等

結果

batch-size = 18時, 一個batch平均用時:
image 載入和預處理 : 0.4849s;
網路前向+損失計算+後傳用時:0.04308s

結論

1.網路過小時, CPU和GPU用時比例不合理,可能需要更大的網路才能合理利用GPU
2.需要處理器的效能很強才行
3.預處理部分不知道能不能在GPU上執行