在联邦学习上下文中,就像本教程所示,全局模型的初始权重(在服务器级别)随机初始化为: state = iterative_process.initialize()
。我想通过从另一个模型(load_model()
)下载这些初始权重来放置这些初始权重。那么请问我该如何继续,我是TFF的新手。谢谢
问问题
68 次
1 回答
1
您可以手动创建具有相同结构和您选择的任何值的状态对象,也可以使用tff.structure.update_struct
.
如果你已经有一个my_model
, 和一个实例或者tff.learning.Model
你tf.keras.Model
想要的权重,你可以写类似的东西
state = tff.structure.update_struct(state, model=tff.learning.ModelWeights.from_model(my_model))
于 2021-03-10T07:59:37.543 回答