Can someone please give me the code for VAE using cifar10 dataset?
only using pytorch. I have written the code with MNIST dataset. need it with cifar

Comments

arhetorical t1_iz55woo wrote on December 6, 2022 at 3:46 PM

Can you not just change the dimensions to work with CIFAR10?

Naive_Weird5939 OP t1_iz56im7 wrote on December 6, 2022 at 3:50 PM

no man it isnt working i have tried every way

gives out error

It doesn’t have Vae using cifar10

It's an image database just use any of the convolutional variational autoencoders

Have gone through it the network used is different

Literally the first one is applicable https://github.com/AntixK/PyTorch-VAE