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.

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

Introduce flower-simulation command and update simulation examples #3009

Merged
merged 137 commits into from Feb 29, 2024

Conversation

jafermarq
Copy link
Contributor

@jafermarq jafermarq commented Feb 25, 2024

Run simulation with FlowerNext in a single command. SuperLink is terminated once the ServerApp finishes.

  • Updates simulation examples so they run in both start_simulation and flower-simulation mode:

    • PyTorch simulation example
    • Tensorflow simulation example <-- #TODO: enable GPU growth for ServerApp
    • Make TF work with flower-simulation launch

Test by:

cd examples/simulation-pytorch
flower-simulation --client-app=sim:client --server-app=sim:server --num-supernodes=100

or:

cd examples/simulation-tensorflow
# append to below `--enable-tf-gpu-growth` if you have a GPU
flower-simulation --client-app=sim:client --server-app=sim:server --num-supernodes=100

Merge first: #3031

Base automatically changed from vce-fleet-api-loop to main February 28, 2024 17:15
@jafermarq jafermarq marked this pull request as ready for review February 29, 2024 11:46
@danieljanes danieljanes enabled auto-merge (squash) February 29, 2024 20:59
@danieljanes danieljanes merged commit f25ddc1 into main Feb 29, 2024
28 checks passed
@danieljanes danieljanes deleted the vce-fleet-run-simulation branch February 29, 2024 21:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Examples Add or update a Flower example simulation An issue related to Flower's simulation
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants