tensorflow-federated - 如何在 tensorflow 联合中获取客户端模型（基于 Keras）？

Question

我正在寻找一种将客户端模型作为检查点的方法，我可以将其作为标准 keras 模型进行调查。我研究了这个问题，但它只提供了权重，有没有办法在联合训练期间直接从客户那里获取或保存模型？

Answer 1

原始问题中引用的答案提供了一种发回模型权重的方法；似乎缺少的是如何将这些权重分配给tf.keras.Model具有相同架构的？使用tff.learning.assign_weights_to_keras_model，可以使用客户端权重初始化 Keras 模型实例，并且该实例将提供标准的 Keras API（evaluate()等test_on_batch()）。

文本生成的联邦学习教程有一个这样做的例子。