在 TFF 中：如何保持（或保存）表现最好的“状态”

Question

在 TFF 中，变量 state 用于评估，如下所示：

test_metrics = evaluation(state.model, federated_test_data)

那么当我找到性能最好的一个时，我们如何保留或保存这个变量

Answer 1

tensorflow_federated/python/research/utils/checkpoint_manager.py 模块中有用于序列化状态的实用程序。使用CheckpointManager 的示例可以在tensorflow_federated/python/research/utils/training_loop.py 中找到

CheckpointManager 是tf.saved_model 包的一个相当薄的包装。对于您的场景，直接使用它可能更灵活。