Dataloader Iterator Not working on custom datataset

usuyama / pytorch-unet

Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation

https://colab.research.google.com/github/usuyama/pytorch-unet/blob/master/pytorch_unet_resnet18_colab.ipynb

MIT License

827 stars 233 forks source link

Dataloader Iterator Not working on custom datataset #4

Closed gulzainali98 closed 5 years ago

gulzainali98 commented 5 years ago

Kindly go through this whole post https://discuss.pytorch.org/t/dataloader-notimplemented-error/35349/2

I have described series of errors i encountered. Now Kindly let me know what is the issue with dataloader, it is returning "NotImplemented" Exception.

usuyama commented 5 years ago

Looks like the issue is resolved?

So all the problems above mentioned are solved by implementing len(self) function in the dataloader.