Skip to content

Commit 348133c

Browse files
Added few lines of code in Keras Custom model to fix the error
Added 2 lines of code `import keras`, `@keras.saving.register_keras_serializable()` on top of the Keras Custom model definition because it shows typeerror - "Make sure custom classes are decorated with `@keras.saving.register_keras_serializable()`" after reloading the model when saving as `.keras`. As mentioned in this Saving [Custom objects](https://www.tensorflow.org/guide/keras/serialization_and_saving#custom_objects) definition. Please refer to this replicated working code [gist](https://colab.research.google.com/gist/RenuPatelGoogle/f124503c52144c5ed6e9c79cffcb593d/intro_to_modules.ipynb#scrollTo=hpW-4Dj3Q1Rb) for your reference. Thank you.
1 parent c6fe18c commit 348133c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

site/en/guide/intro_to_modules.ipynb

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -918,6 +918,8 @@
918918
},
919919
"outputs": [],
920920
"source": [
921+
"import keras
922+
"@keras.saving.register_keras_serializable()
921923
"class MySequentialModel(tf.keras.Model):\n",
922924
" def __init__(self, name=None, **kwargs):\n",
923925
" super().__init__(**kwargs)\n",

0 commit comments

Comments
 (0)