Hi,
I’d like to train a VAE on a single digit from the MNIST.  Is it possible for the train loader to output only a given class instead of all classes randomly or do I have to write a custom data loader?
Thank you.