zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 04:14:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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
Download Patch File Download Diff File Expand all files Collapse all files

4
cvae/requirements.txt Normal file
View File

@@ -0,0 +1,4 @@
mlx>=0.0.9
mlx-data
numpy
Pillow