How to adapt this code to torchvion.dataset?

chaiyujin / glow-pytorch

pytorch implementation of openai paper "Glow: Generative Flow with Invertible 1×1 Convolutions"

MIT License

505 stars 79 forks source link

How to adapt this code to torchvion.dataset? #28

Open StarKnight7 opened 2 years ago

StarKnight7 commented 2 years ago

I would like to use torchvision.dataset.MNIST to run the code since celebA dataset takes more time to train. Please could you tell me what changes to make if I want to train this code for the MNIST dataset

Moleculebo commented 1 year ago

Hi！do you have solved the problem? can you share your code? thanks!