AttributeError: module 'tensorflow_federated.python.learning' has no attribute 'from_keras_model'

Question

Attempting to run federated learning on tff however, encountering the following:

Code: trainer = tff.learning.algorithms.build_weighted_fed_avg( model_fn, client_optimizer_fn=lambda: tf.keras.optimizers.Adam(), server_optimizer_fn=lambda: tf.keras.optimizers.Adam() )

state = trainer.initialize() train_hist = [] for i in range(EPOCHS): state, metrics = trainer.next(state, train_data) train_hist.append(metrics)

print(f"\rRun {i+1}/{EPOCHS}", end="")

Environment: Using Google collab python - 3.10.12 TensorFlow Federated version: 0.61.0

Any help appreciated.

Attempted to downgrade version of TFF

Raihanul Alam Hridoy · Accepted Answer · 2024-11-09 15:10:34Z

0

Now "from_keras_model" is moved to "models". So, "tff.learning.from_keras_model" is replaced with "tff.learning.models.from_keras_model"

Old Code

  return tff.learning.from_keras_model(
      keras_model,
      input_spec=xxxxx,
      loss=tf.keras.losses.SparseCategoricalCrossentropy(),
      metrics=[tf.keras.metrics.SparseCategoricalAccuracy()])

NEW Code

  return tff.learning.models.from_keras_model(
      keras_model,
      input_spec=xxxxx,
      loss=tf.keras.losses.SparseCategoricalCrossentropy(),
      metrics=[tf.keras.metrics.SparseCategoricalAccuracy()])

edited Nov 9, 2024 at 15:10

Raihanul Alam Hridoy

5772 gold badges8 silver badges26 bronze badges

answered Nov 9, 2023 at 15:08

I Bajwa PHD

1,7661 gold badge21 silver badges45 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

AttributeError: module 'tensorflow_federated.python.learning' has no attribute 'from_keras_model'

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related