save checkpoint as .weights.h5

Author: mahsanghani

site/en/tutorials/distribute/keras.ipynb

@@ -363,7 +363,7 @@
         "# Define the checkpoint directory to store the checkpoints.\n",
         "checkpoint_dir = './training_checkpoints'\n",
         "# Define the name of the checkpoint files.\n",
-        "checkpoint_prefix = os.path.join(checkpoint_dir, \"ckpt_{epoch}\")"
+        "checkpoint_prefix = os.path.join(checkpoint_dir, \"ckpt_{epoch}.weights.h5\")"
       ]
     },
     {
@@ -396,7 +396,7 @@
         "# Define a callback for printing the learning rate at the end of each epoch.\n",
         "class PrintLR(tf.keras.callbacks.Callback):\n",
         "  def on_epoch_end(self, epoch, logs=None):\n",
-        "    print('\\nLearning rate for epoch {} is {}'.format(        epoch + 1, model.optimizer.lr.numpy()))"
+        "    print('\\nLearning rate for epoch {} is {}'.format(epoch + 1, model.optimizer.lr.numpy()))"
       ]
     },
     {