About the args in FedML parrot examples.
See original GitHub issuefrom the step3 in docs https://doc.fedml.ai/simulation/examples/sp_fedavg_mnist_lr_example.html, I am told to execute the following command to run the example code:
python torch_fedavg_mnist_lr_one_line_example.py --cf fedml_config.yaml
However, when I tried to modify the args in this YAML file (eg. set the using_gpu to true), I found that in the runtime the training is still based on the CPU.
So I checked the code in fedml/lib/python3.7/site-packages/fedml/arguments.py, line 63, and I found a snippet of code as follows:
`
path_current_file = path.abspath(path.dirname(__file__))
if training_type == "simulation" and comm_backend == "single_process":
config_file = path.join(path_current_file, "config/simulation_sp/fedml_config.yaml")
cmd_args.yaml_config_file = config_file
elif training_type == "simulation" and comm_backend == "MPI":
config_file = path.join(
path_current_file, "config/simulaton_mpi/fedml_config.yaml"
)
cmd_args.yaml_config_file = config_file
elif training_type == "cross_silo":
pass
elif training_type == "cross_device":
pass
else:
pass
` It seems during the simulation, it does not matter how you set the YAML file, the default one would be loaded.
Issue Analytics
- State:
- Created a year ago
- Reactions:1
- Comments:12 (5 by maintainers)
Top Results From Across the Web
FedML Parrot User Guide
FedML Parrot can smoothly transplant your simulation code to real-world deployment with our ... Benchmark. Config MLOps parameters. Login to MLOps. Examples ......
Read more >FedML/README.md at master - GitHub
FedML Parrot - Simulating federated learning in the real world (1) simulate FL ... centralized: Some centralized trainer code examples for ...
Read more >FLUTE: A Scalable, Extensible Framework for High ... - arXiv
A sample of the platform capabilities is also presented for a range of tasks, ... the leading FL simulation platforms2, i.e. FedML (He...
Read more >fedml - PyPI
simulation: FedML parrot can support: (1) simulate FL using a single ... centralized: Some centralized trainer code examples for benchmarking purposes.
Read more >A Review of Machine Learning for Big Data Analysis
(FLOPs) and 54-105 times fewer parameters in the edge CNN than FedAvg (He et ... in [2021], FedML systems were used as a...
Read more >
Top Related Medium Post
No results found
Top Related StackOverflow Question
No results found
Troubleshoot Live Code
Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free
Top Related Reddit Thread
No results found
Top Related Hackernoon Post
No results found
Top Related Tweet
No results found
Top Related Dev.to Post
No results found
Top Related Hashnode Post
No results found
@JLU-Neal @mlpotter I’ve optimized the source code and solved your issues. Please pip install fedml==0.7.27
@JLU-Neal OK. The issue is super outdated and we iterated a lot in the past few months. Please switch to our new version later if you still need to use FedML.