search

securefederatedai / openfl

An Open Framework for Federated Learning.
https://openfl.readthedocs.io/en/latest/index.html
Apache License 2.0
734 stars 207 forks source link

Add pytorch MNIST Workflow tutorial with FedProx #1158

Closed ynonflumintel closed 1 week ago

ynonflumintel commented 1 week ago

This branch introduces a tutorial for setting up a neural network training federation using pytorch, the MNIST dataset using the openfl workflow API.

rahulga1 commented 1 week ago

@ynonflumintel I think this is not the correct file in this commit, I dont see workflow api getting used in this file. can you please confirm?

ynonflumintel commented 1 week ago

@ynonflumintel I think this is not the correct file in this commit, I dont see workflow api getting used in this file. can you please confirm?

I pushed the wrong version of the file before, please take a look now

ynonflumintel commented 1 week ago

General comment on the entire notebook: Include output of all the cells. New users should have visibility into the cells and evaluate for themselves if the notebook fulfils their goal of visit. Not showing output, can put-off newcomers because they end goal is not "visible".

Thought it would be too verbose and increase the file size significantly, but I added them as per your request