search

pclucas14 / pixel-cnn-pp

Pytorch Implementation of OpenAI's PixelCNN++
Other
345 stars 76 forks source link

Train for my own dataset #18

Closed NeroArcher closed 4 years ago

NeroArcher commented 4 years ago

Thank you for your advice last time, I can run the trainning process properly after that. My next goal is to train it with my own dataset(probably hundreds of images) instead of Mnist or Cifar. I have read something about the dataloader part but I am still a little confused about where and how I should do the adjusting in the code. May I get any more advice from you?

pclucas14 commented 4 years ago

It's hard to say with this little information. Probably best to build your own Pytorch Dataset , then change places in the code where the data size is hardcoded. After that hopefully it works.

NeroArcher commented 4 years ago

It's hard to say with this little information. Probably best to build your own Pytorch Dataset , then change places in the code where the data size is hardcoded. After that hopefully it works.

Thank you! I will learn from there!