Use legacy optimizer because the optimizer is applied on two models.

PiperOrigin-RevId: 476931964

Author: chenmoneygithub

site/en/tutorials/generative/style_transfer.ipynb

@@ -89,6 +89,13 @@
         "This is implemented by optimizing the output image to match the content statistics of the content image and the style statistics of the style reference image. These statistics are extracted from the images using a convolutional network."
       ]
     },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "28W8ggyO1KER"
+      },
+      "source": []
+    },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -1029,7 +1036,7 @@
         "id": "lcLWBQChsutQ"
       },
       "source": [
-        "Reinitialize the optimization variable:"
+        "Reinitialize the image-variable and the optimizer:"
       ]
     },
     {
@@ -1040,6 +1047,7 @@
       },
       "outputs": [],
       "source": [
+        "opt = tf.keras.optimizers.Adam(learning_rate=0.02, beta_1=0.99, epsilon=1e-1)\n",
         "image = tf.Variable(content_image)"
       ]
     },