How to undo @tf.keras.utils.register_keras_serializable

Viewed 294

You use this to register a func into keras serialization framework. I did this in a notebook cell:

@tf.keras.utils.register_keras_serializable()
def foo():
  return tf.constant(1)

But if I need to make changes to the method, you will get a ValueError: [...] has already been registered to [...].

Is there a way to unregister whatever that was, and then re-register with the updated func?

0 Answers
Related