Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge


Converts a tf.keras.Model to a tff.learning.models.FunctionalModel.

keras_model A tf.keras.Model object, should be uncompiled. If compiled, the metrics, optimizer, and loss function will be ignored. Note: models that have multiple outputs will send all outputs to the loss_fn.
loss_fn A tf.keras.losses.Loss object.
input_spec A structure of tf.TensorSpec defining the input to the model.

A tff.learning.models.FunctionalModel.

KerasFunctionalModelError the model has a batch normalization layer.