apple / pfl-research

Simulation framework for accelerating research in Private Federated Learning
http://apple.github.io/pfl-research/
Apache License 2.0
281 stars 27 forks source link

Implement TF setup for FLAIR #90

Open grananqvist opened 1 month ago

grananqvist commented 1 month ago

FLAIR is only available in TFF or pfl+PyTorch. We need to provide a setup for pfl+TF. Should have parity in performance with pfl+PyTorch (see table 2 in paper https://arxiv.org/pdf/2404.06430).

see e.g. image_classification how both pytorch and tf has identical setups.

(120904681)