Skip to content

Commit 3f71a51

Browse files
authored
Update README.md
1 parent 1e52219 commit 3f71a51

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,3 +9,4 @@ To run our code, colab is required. Upload the folder into google drive and then
99
* `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.
1010
* `beta_vae_quickdraw.ipynb` - notebook for our beta-VAE trained on the Quick Draw dataset.
1111
* `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

Comments
 (0)