Merge pull request #1166 from Azure/minxia/patch

patch for resume training notebook
2025-12-19 17:17:04 -05:00 · 2020-09-29 17:30:24 -07:00
parent 1d0766bd46 89c35c8ed6
commit 9fe459be79
1 changed files with 4 additions and 2 deletions
--- a/how-to-use-azureml/ml-frameworks/tensorflow/train-tensorflow-resume-training/train-tensorflow-resume-training.ipynb
+++ b/how-to-use-azureml/ml-frameworks/tensorflow/train-tensorflow-resume-training/train-tensorflow-resume-training.ipynb
@@ -420,7 +420,9 @@
        "                      script='tf_mnist_with_checkpoint.py',\n",
        "                      arguments=args,\n",
        "                      compute_target=compute_target,\n",
-        "                      environment=tf_env)"
+        "                      environment=tf_env)\n",
+        "\n",
+        "src.run_config.data_references = {checkpoint_data_ref.data_reference_name : checkpoint_data_ref.to_config()}"
      ]
    },
    {
@@ -497,4 +499,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}