Merge pull request #2278 from RenuPatelGoogle:patch-27

PiperOrigin-RevId: 574308985
tensorflow · Oct 18, 2023 · 008b211 · 008b211
2 parents d742984 + a3c2132
commit 008b211
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/site/en/guide/intro_to_modules.ipynb b/site/en/guide/intro_to_modules.ipynb
@@ -91,6 +91,7 @@
       "outputs": [],
       "source": [
         "import tensorflow as tf\n",
+        "import keras\n",
         "from datetime import datetime\n",
         "\n",
         "%load_ext tensorboard"
@@ -918,6 +919,7 @@
       },
       "outputs": [],
       "source": [
+        "@keras.saving.register_keras_serializable()\n",
         "class MySequentialModel(tf.keras.Model):\n",
         "  def __init__(self, name=None, **kwargs):\n",
         "    super().__init__(**kwargs)\n",