我已经阅读了 tensorflow.org 上关于 tensorflow federated 的所有文档,但我不确定如何实现我自己的联合算法。例如,我有一个编译好的 keras 模型,我知道如何将其转换为 tff.computation。似乎为了构建一个联合算法,应该构建一个 iterative_process。任何人都可以在这方面帮助我吗?
太感谢了,
我已经阅读了 tensorflow.org 上关于 tensorflow federated 的所有文档,但我不确定如何实现我自己的联合算法。例如,我有一个编译好的 keras 模型,我知道如何将其转换为 tff.computation。似乎为了构建一个联合算法,应该构建一个 iterative_process。任何人都可以在这方面帮助我吗?
太感谢了,
您可以在 tensorflow federated repo 中阅读文件optimizer_utils.py,该文件位于python/learning/framework/
更多详情,请看链接
one_round_computation 函数在一轮中实现了联邦算法的编排逻辑。最重要的是使用 tff.federated_map、tff.federated_broadcast、tff.federated_collect 构建抽象语法树...
您应该使用以下代码:
tff.learning.from_compiled_keras_model(kera_model, dummy_batch)