Pytorch(五)入門:DataLoader 和 Dataset

DataLoader 和 Dataset 構建模型的基本方法,咱們瞭解了。 接下來,咱們就要弄明白怎麼對數據進行預處理,而後加載數據,咱們之前手動加載數據的方式,在數據量小的時候,並無太大問題,可是到了大數據量,咱們須要使用 shuffle, 分割成mini-batch 等操做的時候,咱們可使用PyTorch的API快速地完成這些操做。git Dataset是一個包裝類,用來將數據包裝爲Datas
相關文章
相關標籤/搜索