batchsize大小對訓練速度的影響

1.batchsize越大 是不是訓練越快? GPU :一塊2080Ti 平臺:pytorch cuda 圖片數量:2700 batchsize 一個圖片處理時間 GPU內存佔用 GPU算力使用 一個epoch 所用時間 1 0.117s 2.5G 20% 2700 * 0.0117 = 318s 5 0.516s 8G 90% 2700 * 0.516/5 = 279s batchsize大了之
相關文章
相關標籤/搜索