fedavg.pytorch
fedavg.pytorch copied to clipboard
A PyTorch implementation of "Communication-Efficient Learning of Deep Networks from Decentralized Data", AISTATS, 2017
Original implementation averages all (K) clients instead of randomly selected fraction (C*K) of clients. So I modified this code to pass 'selected_clients' with proper aggregation weights
Bumps [torch](https://github.com/pytorch/pytorch) from 1.4.0 to 2.2.0. Release notes Sourced from torch's releases. PyTorch 2.2: FlashAttention-v2, AOTInductor PyTorch 2.2 Release Notes Highlights Backwards Incompatible Changes Deprecations New Features Improvements Bug fixes...