# x,y为训练输入,和标准输出
train_dataset = torch.utils.data.TensorDataset(x, y)
train_dataLoader = Data.DataLoader(train_dataset, batch_size=64, shuffle=True, num_workers=0)
for step, (images, labels) in enumerate(train_dataLoader):
#即可获得images,labels
自定义数据集 输入输出打包
最新推荐文章于 2023-07-10 20:06:52 发布