Hello, I was following the Pytorch tutorial of [customizing dataloader class], everything works just fine except when running the last segment of code. It just stay there without finishing, and does not print any output either.

tutorialoutput1137×1000 95.1 KB

(Developing Custom PyTorch Dataloaders — PyTorch Tutorials 1.7.1 documentation)