return local metadata from training

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

return local metadata from training #71

Closed grananqvist closed 3 months ago

grananqvist commented 3 months ago

to make the helper fn do_multiple_epochs_of more widely useful for other algorithms, return a new simple dataclass with any information we may want to propagate. Only number of local steps taken is known to be useful at this moment.

rdar://124202046