如何保存Tensorflow联合模型

问题描述

如何轻松保存Tensorflow联合模型？ （状态）

几个月前，我在导入 ServerState 和 FileCheckPointManager 之后使用了此解决方案，并且有效：

import datetime

date_object = datetime.date.fromisoformat('2021-01-01')

foo = Foo(bar=date_object)

但是现在此解决方案不再起作用，因为Documentation不再包含 # Create the checkpoint Manager ckpt_manager = FileCheckpointManager(root_dir=checkpoint_dir) # Save checkpoint for round N ckpt_manager.save_checkpoint(ServerState.from_tff_result(state),round_num=NUM_ROUNDS) methotd。

from_tff_result

我还使用了包含metohd的旧版ServerState，我得到了：

AttributeError: type object 'ServerState' has no attribute 'from_tff_result'

如何轻松保存联盟模型？

解决方法

在进行recent change阻止Struct（以前是AnonymousTuple）返回给用户之后，不再需要{ {1}}； ServerState.from_tff_result(state)应该已经成为一个state。

TLDR：如果将示例更新为简单阅读：

ServerState

它应该可以正常工作。