-
Notifications
You must be signed in to change notification settings - Fork 798
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
refactor(examples) Update quickstart-mlx
example
#3701
base: main
Are you sure you want to change the base?
Conversation
Co-authored-by: Daniel J. Beutel <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It looks very good! I love it! Should we add module docstrings?
ok! What would you propose? something minimal as what's generated by the templates? |
examples/quickstart-mlx/README.md
Outdated
@@ -129,30 +107,30 @@ data = ( | |||
test_partition["label"].astype(np.uint32), | |||
) | |||
|
|||
train_images, train_labels, test_images, test_labels = map(mlx.core.array, data) | |||
train_images, train_labels, test_images, test_labels = map(mx.array, data) | |||
``` | |||
|
|||
### The model |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@charlesbvll how about if we remove the model block. in the end this is not so Flower-specific...
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think we could remove the code (so we don't have to maintain the same code in 2 different places), but it might be good to keep some explanation of what the model is and what it is doing to give some context
Update
examples/quickstart-mlx
to make itflwr run
-compatible. Code generated viaflwr new
, then adapted to the newclient_fn
logic.