Skip to content

Commit 008b211

Browse files
Merge pull request #2278 from RenuPatelGoogle:patch-27
PiperOrigin-RevId: 574308985
2 parents d742984 + a3c2132 commit 008b211

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
@@ -91,6 +91,7 @@
9191
"outputs": [],
9292
"source": [
9393
"import tensorflow as tf\n",
94+
"import keras\n",
9495
"from datetime import datetime\n",
9596
"\n",
9697
"%load_ext tensorboard"
@@ -918,6 +919,7 @@
918919
},
919920
"outputs": [],
920921
"source": [
922+
"@keras.saving.register_keras_serializable()\n",
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)