в TFF: как сохранить (или сохранить) «состояние» наиболее эффективных

Question

в TFF, переменная state используется для оценки, как показано ниже:

test_metrics = evaluation(state.model, federated_test_data)

Итак, как мы можем сохранить эту переменную или сохранить ее, когда я найду лучшую из самый эффективный

Answer 1

В модуле tenorflow_federated / python / research / utils / checkpoint_manager.py имеются утилиты для сериализации состояния. Пример использования CheckpointManager можно найти в tenorflow_federated / python / research / utils / training_l oop .py

CheckpointManager - довольно тонкая оболочка вокруг tf.saved_model упаковка. Использование этого напрямую может быть более гибким для вашего сценария.