已解决：使用Flower和Pytorch模拟联合学习语义分割任务时未加载模型参数

6ovsh4lw 于 2023-05-07 发布在其他

关注(0)|答案(1)|浏览(151)

我正在使用flower和Pytorch联合一个UNET语义分割工作流。到目前为止，我可以加载数据并运行集中式训练，但一旦我尝试联合它，我就会看到模型参数没有正确加载。我在代码和日志输出中包含了一个google colab notebook，以保持问题简短。
https://colab.research.google.com/drive/1dmlH4QTX_ZwicbSfwVeCw55BXRnV6PY4?usp=sharing
我把这个留在这里，以防有人试图实现类似的工作流程。请随意联系。

pytorch

来源：https://stackoverflow.com/questions/76140070/resolved-model-parameters-not-loading-when-simulating-a-federated-learning-sema

1条答案

按热度按时间

qv7cva1a1#

愚蠢的错误，但我发现我试图打电话

def set_parameters(device, net, parameters: List[np.ndarray]): 
  params_dict = zip(net.state_dict().keys(), parameters)
  state_dict = OrderedDict({k: torch.Tensor(v) for k, v in params_dict})
  net.load_state_dict(state_dict, strict=True)

它在客户端类之外，所以我把它移到了类中，现在模型参数正在初始化。

赞(0）回复(0）举报 2023-05-07

我来回答

已解决：使用Flower和Pytorch模拟联合学习语义分割任务时未加载模型参数

1条答案

相关问题

热门标签

最新问答