FLamby
FLamby copied to clipboard
Add proper testing for AdaGrad, Yogi and FedAvg based on FedAdam testing
It would be nice if one client only testing could be used with allclose
statements to check perfect equality between federated versions and their torch.optim counterparts
.
This is done for now only for FedAdam in tests/strategies/test_fed_adam.py