tensorflow - 在 TensorFlow Federated 中使用 update_struct 函数时出错

Question

我正在尝试运行 Minimal Stand-Alone Implementation of Federated Averaging来自 TensorFlow Federated GitHub 存储库，但在 server_update 函数中收到以下错误:
AttributeError:模块“tensorflow_federated.python.common_libs.structure”没有属性“update_struct”
我有一些旧的 TensorFlow Federated 代码使用 tff.utils 包中的 update_state 函数代替 update_struct() 但根据 GitHub 上的提交，这个包现在是空的。我正在使用 TensorFlow Federated 版本 0.18.0，在 Google CoLab 上尝试时也遇到了同样的问题。
我的问题是如何解决这个错误？
谢谢，任何帮助表示赞赏。

Answer 1

我假设您遇到了您描述的错误 here .
似乎该符号不在 0.18 版本中。您可以依赖每晚版本( pip install tensorflow-federated-nightly )，或者修改该行以直接构造新对象，而不是使用 update_struct helper 。也就是说，链接的命令可以更改为:

return ServerState(model_weights,
                   server_optimizer.variables(),
                   server_state.round_num + 1))