Fix suptitles.

PiperOrigin-RevId: 447842364

Author: MarkDaoust

site/en/tutorials/images/classification_with_model_garden.ipynb

@@ -251,7 +251,7 @@
         "  print('This may be broken in Colab.')\n",
         "  device = 'TPU'\n",
         "else:\n",
-        "  print('This is slow, and doesn\\'t train to convergence.')\n",
+        "  print('Running on CPU is slow, so only train for a few steps.')\n",
         "  device = 'CPU'\n",
         "\n",
         "if device=='CPU':\n",
@@ -261,7 +261,6 @@
         "  train_steps=5000\n",
         "  exp_config.trainer.steps_per_loop = 100\n",
         "\n",
-        "exp_config.trainer.steps_per_loop = 100\n",
         "exp_config.trainer.summary_interval = 100\n",
         "exp_config.trainer.checkpoint_interval = train_steps\n",
         "exp_config.trainer.validation_interval = 1000\n",
@@ -582,7 +581,7 @@
         "show_batch(images, labels, tf.cast(predictions, tf.int32))\n",
         "\n",
         "if device=='CPU':\n",
-        "  plt.title('The model was only trained for a few steps, so it is not expected to do well.')"
+        "  plt.suptitle('The model was only trained for a few steps, it is not expected to do well.')"
       ]
     },
     {
@@ -669,7 +668,7 @@
         "  show_batch(data['image'], data['label'], predictions)\n",
         "\n",
         "  if device=='CPU':\n",
-        "    plt.title('The model was only trained for a few steps, it is not expected to do well.')"
+        "    plt.suptitle('The model was only trained for a few steps, it is not expected to do well.')"
       ]
     }
   ],