cloneofsimo / minDiffusion

Self-contained, minimalistic implementation of diffusion models with Pytorch.
866 stars 122 forks source link

how to load custom dataset? #8

Open LexiestLeszek opened 1 year ago

LexiestLeszek commented 1 year ago

Hi! Im new to this stuff and don't understand how can i download a custom dataset and how to add it to this model? My questions are: what format should downloaded dataset be and how to train the model on it?