Update train-tensorflow-resume-training.ipynb

2020-09-29 17:27:23 -07:00
parent 1d0766bd46
commit 33168c7f5d
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",
+        "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
-}
+}