torch.utils.data.DataLoader(image_datasets[x],
batch_size=batch_size,
shuffle=True,
num_workers=8,
pin_memory=True)
num_workers=8:设置线程数
pin_memory=True:由CPU传输的数据不需要经过RAM,直接映射到GPU上。