jjzjj

DataLoaders

全部标签

python - PyTorch:如何将 DataLoaders 用于自定义数据集

如何使用torch.utils.data.Dataset和torch.utils.data.DataLoader根据您自己的数据(不仅仅是torchvision.datasets)?有没有办法使用内置的DataLoaders他们在TorchVisionDatasets上使用用于任何数据集? 最佳答案 是的,这是可能的。只需自己创建对象,例如importtorch.utils.dataasdata_utilstrain=data_utils.TensorDataset(features,targets)train_loader=dat