1

在联邦学习上下文中,就像本教程所示,全局模型的初始权重(在服务器级别)随机初始化为: state = iterative_process.initialize()。我想通过从另一个模型(load_model())下载这些初始权重来放置这些初始权重。那么请问我该如何继续,我是TFF的新手。谢谢

4

1 回答 1

1

您可以手动创建具有相同结构和您选择的任何值的状态对象,也可以使用tff.structure.update_struct.

如果你已经有一个my_model, 和一个实例或者tff.learning.Modeltf.keras.Model想要的权重,你可以写类似的东西

state = tff.structure.update_struct(state, model=tff.learning.ModelWeights.from_model(my_model))

于 2021-03-10T07:59:37.543 回答