Skip to content

Implementation of Federated Learning algorithms such as FedAvg, FedAvgM, SCAFFOLD, FedOpt, Mime using PyTorch.

License

Notifications You must be signed in to change notification settings

BalajiAI/Federated-Learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Federated-Learning

This repository contains the code for Federated Learning algorithms such as FedAvg, FedAvgM, FedAdagrad, FedAdam, FedYogi, SCAFFOLD, MimeLite which are all implemented using PyTorch.

Execution

You can run this code by executing python main.py command in your terminal. You can modify the config.json file to match the set of hyperarameters that you wish to use for execution.

About

Implementation of Federated Learning algorithms such as FedAvg, FedAvgM, SCAFFOLD, FedOpt, Mime using PyTorch.

Resources

License

Stars

Watchers

Forks

Languages