Example of a Convolutional Variational Autoencoder (CVAE) on MNIST (#264)

Browse Source

* initial commit

* style fixes

* update of ACKNOWLEDGMENTS

* fixed comment

* minor refactoring; removed unused imports

* added cifar and cvae to top-level README.md

* removed mention of cuda/mps in argparse

* fixed training status output

* load_weights() with strict=True

* pretrained model update

* fixed imports and style

* requires mlx>=0.0.9

* updated with results using mlx 0.0.9

* removed mention of private repo

* simplify and combine to one file, more consistency with other exmaples

* few more nits

* nits

* spell

* format

---------

Co-authored-by: Awni Hannun <awni@apple.com>

...

This commit is contained in:

Markus Enzweiler

2024-02-07 05:02:27 +01:00

committed by GitHub

parent 8071aacd98

commit 9b387007ab

11 changed files with 531 additions and 2 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

BIN

cvae/assets/rec_mnist.png Normal file

Binary file not shown.

Side by Side After Width:  |  Height:  |  Size: 136 KiB