lucidrains / imagen-pytorch

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch
MIT License
8.09k stars 768 forks source link

how ta activate shuffle for DataLoader? #361

Open TheZaind opened 1 year ago

TheZaind commented 1 year ago

Yeah, the question is in the title. How can I shuffle my Trainingdata when I am training the model? Normal, It's something like : "'shuffle': True" or "data_loader = DataLoader(dataset, batch_size=batch_size, shuffle=True)" but i dnt got a DataLoader in this way. But I don't know where and how to add this. Anyone an idea? Thanks 👍