PyTorch 有没有把 Dataloader 的数据快速转换到 "cuda:0" 的方法?

Question

gpu训练较快，对代码进行改造，其它都改好了，只剩 dataloader 了

# 创建
dataset = Dataset_twin(path_txt_train,transform=transforms.Compose([transforms.Resize((224,224)),transforms.ToTensor()]))
dataLoader = DataLoader(dataset=dataset,batch_size=50,drop_last=True,shuffle=True)

# 使用
for ((img,shape,Y,location),(img_2,shape_2,Y_2,location_2),is_same) in dataLoader:
    print(shape_2.size())
    break

返回的数据里面包含元组，我不想把他们全部打散，有解决办法吗（如题）？

Answer 1

无，一个一个转