Pytorch入門---學習筆記五---Dataloader與Dataset

模型訓練五部曲：數據、模型、損失函數、優化器、迭代訓練 一 數據 1.Dataloader torch.utils.data.Dataloader 功能：構建可迭代帶數據裝載器 dataset:Dataset類，決定數據從哪讀取及如何讀取 batchsize:批大小 num_works：是否多進程讀取數據 shuffle：每個epoch是否亂 drop_last:當樣本數不能被batchsize整

>>阅读原文<<