diff --git a/training/distributed_training/tensorflow/data_parallel/efficientnet/tensorflow2_smdataparallel_efficientnet_demo.ipynb b/training/distributed_training/tensorflow/data_parallel/efficientnet/tensorflow2_smdataparallel_efficientnet_demo.ipynb
index 088fa3bb6d..a63d523198 100644
--- a/training/distributed_training/tensorflow/data_parallel/efficientnet/tensorflow2_smdataparallel_efficientnet_demo.ipynb
+++ b/training/distributed_training/tensorflow/data_parallel/efficientnet/tensorflow2_smdataparallel_efficientnet_demo.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "28a189bb",
+   "id": "db45f0c5",
    "metadata": {},
    "source": [
     "# Distributed Data Parallel EfficientNet Training with TensorFlow2 and SageMaker Distributed\n",
@@ -29,7 +29,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "b70b048c",
+   "id": "62efb0c1",
    "metadata": {},
    "source": [
     "## Amazon SageMaker Initialization\n",
@@ -48,7 +48,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "cb9a80f5",
+   "id": "94d8089f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -80,7 +80,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "bd2a260e",
+   "id": "76353ab4",
    "metadata": {},
    "source": [
     "To verify that the role above has required permissions:\n",
@@ -94,7 +94,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "a72c90b3",
+   "id": "9836f45e",
    "metadata": {},
    "source": [
     "## Prepare SageMaker Training Images\n",
@@ -110,7 +110,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "aca5b15a",
+   "id": "fb5b7fa7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -121,7 +121,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "2622b8f6",
+   "id": "ce70c7c8",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -131,7 +131,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "10d0a52d",
+   "id": "c712784f",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -141,7 +141,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7ee95615",
+   "id": "5f00e075",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -153,7 +153,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f12251c7",
+   "id": "ec8919bc",
    "metadata": {},
    "source": [
     "## Preparing FSx Input for SageMaker\n",
@@ -171,7 +171,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "caf876e9",
+   "id": "8ace4f98",
    "metadata": {},
    "source": [
     "## SageMaker TensorFlow Estimator function options\n",
@@ -202,7 +202,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "884c07bf",
+   "id": "d2bbed39",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -214,7 +214,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "19ade17a",
+   "id": "1832a42e",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -224,7 +224,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0b831ffb",
+   "id": "6123f021",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -243,7 +243,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "91f2e50f",
+   "id": "61391941",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -270,7 +270,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "8d3312b5",
+   "id": "161d67cd",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -285,7 +285,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "60090fa8",
+   "id": "aad83355",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -293,7 +293,7 @@
     "    entry_point=\"main.py\",\n",
     "    role=role,\n",
     "    image_uri=docker_image,\n",
-    "    source_dir=\"./tensorflow/efficientnet\",\n",
+    "    source_dir=\"./SMDDP-Examples/tensorflow/efficientnet\",\n",
     "    instance_count=instance_count,\n",
     "    instance_type=instance_type,\n",
     "    framework_version=\"2.6\",\n",
@@ -311,7 +311,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "ee3094b5",
+   "id": "c9a7eafa",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -335,7 +335,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "4ff562d8",
+   "id": "e34fbd06",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -345,7 +345,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "44c4f230",
+   "id": "8ccab7b1",
    "metadata": {},
    "source": [
     "## Next steps\n",
@@ -356,7 +356,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "0438d710",
+   "id": "dbaa9162",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -367,7 +367,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "82857412",
+   "id": "5963f464",
    "metadata": {},
    "source": [
     "## Clean Up\n",