在深度学习中,我们会发现数据集通常会转载到DataLoader中,
dataloader = DataLoader(datesets, batch_size=64, shuffle=True)
当然这里的datesets可以指整个数据集,也可以是训练集,测试集等
#装载整个数据集
datasets = {
'train':train_ds ,
'valid':valid_ds ,
'test': test_ds
}
dataLoader = {
ds: DataLoader(datasets[ds],
batch_size=</