TFF: вручную установить начальное состояние

Question

Пожалуйста, я хотел бы изменить этот код , нажав начальные веса state.model которые генерируются случайным образом ( state = iterative_process.initialize() ) другими весами, загруженными из внешнего файла:

      def create_keras_model():
  return tf.keras.models.Sequential([
      tf.keras.layers.InputLayer(input_shape=(784,)),
      tf.keras.layers.Dense(10, kernel_initializer='zeros'),
      tf.keras.layers.Softmax(),
  ])
def model_fn():
  keras_model = create_keras_model()
  return tff.learning.from_keras_model(
      keras_model,
      input_spec=preprocessed_example_dataset.element_spec,
      loss=tf.keras.losses.SparseCategoricalCrossentropy(),
      metrics=[tf.keras.metrics.SparseCategoricalAccuracy()])

iterative_process = tff.learning.build_federated_averaging_process(
    model_fn,...)

state = iterative_process.initialize()

Эта последняя строка state = iterative_process.initialize() который выражает случайную генерацию начальных весов, как я могу его изменить?