Code

Update apfl client#62

Code

Merged

jewelltaylor merged 14 commits intomainVectorInstitute/FL4Health:mainfrom update-apfl-clientVectorInstitute/FL4Health:update-apfl-clientCopy head branch name to clipboardOct 18, 2023

+414-486

Commits

Commits on Sep 28, 2023

• First pass at changing apfl client to leverage base client

  jewelltaylor

  committed
  2a96ad0

  Copy full SHA for 2a96ad0

Commits on Sep 29, 2023

• Add comments. Add tests to make sure optimizers are being set properly. Make sure update_after_step is being passed the actual_step not just local step in each epoch.

  jewelltaylor

  committed
  41f4d7d

  Copy full SHA for 41f4d7d

• fix pre-commit issues

  jewelltaylor

  committed
  059fbe4

  Copy full SHA for 059fbe4

Commits on Oct 12, 2023

• Remove the need to define fedprox fit function

  jewelltaylor

  committed
  d50ae1e

  Copy full SHA for d50ae1e

Commits on Oct 13, 2023

• Add MetricMeterManager to handle one or more MetricMeters. Define custom compute loss for APFL. Return dict with personal, global and local preds from ApflClient.predict. Make some changes to Basic…

  Show description for f462d69
  jewelltaylor

  committed
  f462d69

  Copy full SHA for f462d69

• Have get optimizer return optimizer or dict of optimizer. Update apfl examlples accordingly. Remove no longer relevant split optimizer test

  jewelltaylor

  committed
  61693e7

  Copy full SHA for 61693e7

• Slight typing change

  jewelltaylor

  committed
  6b75b74

  Copy full SHA for 6b75b74

• Merge in changes from main and change the way we increment steps in train_by_epochs

  jewelltaylor

  committed
  349c7cb

  Copy full SHA for 349c7cb

Commits on Oct 16, 2023

• Initialize MetricMeterManager with mapping between prediciton key in Client constructor

  jewelltaylor

  committed
  6cf8386

  Copy full SHA for 6cf8386

• Add some comments as suggested

  jewelltaylor

  committed
  e15e2f8

  Copy full SHA for e15e2f8

• Merge in recent changes from main and resolve conflicts

  jewelltaylor

  committed
  c28e38d

  Copy full SHA for c28e38d

Commits on Oct 17, 2023

• Fix unwanted changes from resolving merge conflict

  jewelltaylor

  committed
  898c274

  Copy full SHA for 898c274

Commits on Oct 18, 2023

• Fix APFL example

  jewelltaylor

  committed
  2498ee5

  Copy full SHA for 2498ee5

• Updates based on Davids suggestions

  jewelltaylor

  committed
  9959c89

  Copy full SHA for 9959c89