Skip to content
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.

Sign up for GitHub

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

Jump to bottom

refactor(framework) Update mlx,tf and pytorch templates #3933

Merged
merged 10 commits into from
Aug 5, 2024
Merged

refactor(framework) Update mlx,tf and pytorch templates #3933

merged 10 commits into from
Aug 5, 2024

Conversation

jafermarq
Copy link
Contributor

@jafermarq jafermarq commented Jul 27, 2024 •
edited
Loading

Several minor updates:

  • Consistent usage of device in pytorch template.
  • Revisited train function in pytorch template.
  • Using smaller model in tf template (comparable to that in pytorch template)
  • Raise flwr and flwr-datasets to latest version in pyprojec.toml
  • Other minor tidy ups

@jafermarq jafermarq marked this pull request as ready for review July 27, 2024 13:58
@danieljanes danieljanes merged commit 6a639c6 into main Aug 5, 2024
33 checks passed
@danieljanes danieljanes deleted the various-template-updates branch August 5, 2024 18:08
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@danieljanes danieljanes danieljanes approved these changes

@tanertopal tanertopal Awaiting requested review from tanertopal tanertopal is a code owner

@charlesbvll charlesbvll Awaiting requested review from charlesbvll

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@jafermarq @danieljanes