вычисление инициализации для построения состояния сервера

Question

В контексте федеративного обучения и, как показано в этом руководстве , начальные веса глобальной модели (на уровне сервера) инициализируются случайным образом с помощью: state = iterative_process.initialize(). Я хочу приложить руку, чтобы поставить эти начальные веса, загрузив их из другой модели ( load_model()). Итак, пожалуйста, как я могу продолжить, я новичок в TFF. Спасибо

Answer 1

Вы можете вручную создать объект состояния с той же структурой и любыми значениями, которые вы выберете, или использовать tff.structure.update_struct.

Если у вас уже есть my_model, и экземпляр любого tff.learning.Model или tf.keras.Model с желаемыми весами вы можете написать что-нибудь вроде

state = tff.structure.update_struct(state, model=tff.learning.ModelWeights.from_model(my_model))