PyTorch Dataset/DataLoader classes - PyTorch Forums
discuss.pytorch.org › t › pytorch-dataset-dataloaderNov 14, 2021 · Hi! When training ResNet on ImageNet dataset, I coded some dataloading functionality by hand, which was extremely useful to me. I am currently transitioning from TF2 to PyTorch and I am very new to PyTorch Dataset and Dataloader classes. I am wondering whether PyTorch Dataset/DataLoader classes make the flow I coded by hand available out of the box. I did read PyTorch tutorials and API docs ...