1

我已经阅读了 tensorflow.org 上关于 tensorflow federated 的所有文档,但我不确定如何实现我自己的联合算法。例如,我有一个编译好的 keras 模型,我知道如何将其转换为 tff.computation。似乎为了构建一个联合算法,应该构建一个 iterative_process。任何人都可以在这方面帮助我吗?

太感谢了,

4

2 回答 2

1

您可以在 tensorflow federated repo 中阅读文件optimizer_utils.py,该文件位于python/learning/framework/

更多详情,请看链接

one_round_computation 函数在一轮中实现了联邦算法的编排逻辑。最重要的是使用 tff.federated_map、tff.federated_broadcast、tff.federated_collect 构建抽象语法树...

于 2020-09-13T13:30:16.093 回答
0

您应该使用以下代码:

tff.learning.from_compiled_keras_model(kera_model, dummy_batch)

于 2020-01-22T09:43:27.050 回答