PyTorch: How to use DataLoaders for custom Datasets

后端 未结 4 1654
梦谈多话
梦谈多话 2020-12-04 07:10

How to make use of the torch.utils.data.Dataset and torch.utils.data.DataLoader on your own data (not just the torchvision.datasets)?<

4条回答
  •  不知归路
    2020-12-04 07:38

    In addition to user3693922's answer and the accepted answer, which respectively link the "quick" PyTorch documentation example to create custom dataloaders for custom datasets, and create a custom dataloader in the "simplest" case, there is a much more detailed dedicated official PyTorch tutorial on how to create a custom dataloader with the associated preprocessing: "writing custom datasets, dataloaders and transforms" official PyTorch tutorial

提交回复
热议问题