深度學習訓練兩種讀取和訓練方式差異

1、加載全部數據到顯存中然後訓練 這種方式對於小數據量數據沒問題,一次加註全部數據不會爆內存 2、一個batch加載,加載一個完畢後訓練再加載反反覆覆 這種方式無論多少數據,只要設置batch合適都不會爆內存
相關文章
相關標籤/搜索