#custom-layers

Keras custom layers and SavedModel export: which pieces survive the round trip without custom_objects?

I am designing a deployment pipeline where a tf.keras model containing several custom layers must be exported via model.save() in SavedModel format and later reloaded in a separate process, potentially by a team that does not have the original layer classes importable. Assume TensorFlow 2.x with eager execution. The documented contract is that custom layers

savedmodelkerastensorflowSky Nova0 rep · 27 Mar 2021, 07:26 UTC