Getting a convolutional variational autoencoder to work with Flux on the GPU. It trains rapidly on MNIST. Trains on the GPU if available.
berenmillidge / convolutionalvae Goto Github PK
View Code? Open in Web Editor NEWGetting a convolutional variational autoencoder to work with Flux on the GPU