You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+1Lines changed: 1 addition & 0 deletions
Original file line number
Diff line number
Diff line change
@@ -9,3 +9,4 @@ To run our code, colab is required. Upload the folder into google drive and then
9
9
*`bvae_classifier_mnist_v3.ipynb` - Five layer fully connected classifier that trains on the encoded representations of the beta-vae (`bVAE.pt`) on the MNIST dataset.
10
10
*`beta_vae_quickdraw.ipynb` - notebook for our beta-VAE trained on the Quick Draw dataset.
11
11
*`vae_classifier_mnist_v3.ipynb` - Five layer fully connected classifier fully connected classifier that trains on the encoded representations of a simple VAE (`vae_epoch_25.pt`) on the MNIST dataset.
12
+
*`hybridvae_classifier_mnist.ipynb` - VAE but with the decoder swapped with a classifier. Model did not train well, so stopped training.
0 commit comments