当前位置: 代码迷 >> 综合 >> torch dataloader
  详细解决方案

torch dataloader

热度:81   发布时间:2024-01-24 09:05:58.0
    dataset=torch.utils.data.TensorDataset(data[0],data[1])loader = torch.utils.data.DataLoader(dataset,batch_size=32,shuffle=True)for index,item in enumerate(loader):print(len(item))print(index)
  相关解决方案