Upgrade the TF2 version to 2.4.1 and also update the py_version to 3.7

Author: karan6181

training/distributed_training/tensorflow/data_parallel/bert/tensorflow2_smdataparallel_bert_demo.ipynb

@@ -244,8 +244,8 @@
     "                        role=role,\n",
     "                        image_uri=docker_image,\n",
     "                        source_dir='deep-learning-models/models/nlp',\n",
-    "                        framework_version='2.3.1',\n",
-    "                        py_version='py3',\n",
+    "                        framework_version='2.4.1',\n",
+    "                        py_version='py37',\n",
     "                        instance_count=instance_count,\n",
     "                        instance_type=instance_type,\n",
     "                        sagemaker_session=sagemaker_session,\n",
@@ -315,4 +315,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}

training/distributed_training/tensorflow/data_parallel/maskrcnn/tensorflow2_smdataparallel_maskrcnn_demo.ipynb

@@ -238,8 +238,8 @@
     "                        role=role,\n",
     "                        image_uri=docker_image,\n",
     "                        source_dir='.',\n",
-    "                        framework_version='2.3.1',\n",
-    "                        py_version='py3',\n",
+    "                        framework_version='2.4.1',\n",
+    "                        py_version='py37',\n",
     "                        instance_count=instance_count,\n",
     "                        instance_type=instance_type,\n",
     "                        sagemaker_session=sagemaker_session,\n",
@@ -323,4 +323,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}

training/distributed_training/tensorflow/data_parallel/mnist/tensorflow2_smdataparallel_mnist_demo.ipynb

@@ -104,7 +104,7 @@
     "                        entry_point='train_tensorflow_smdataparallel_mnist.py',\n",
     "                        role=role,\n",
     "                        py_version='py37',\n",
-    "                        framework_version='2.3.1',\n",
+    "                        framework_version='2.4.1',\n",
     "                        # For training with multinode distributed training, set this count. Example: 2\n",
     "                        instance_count=2,\n",
     "                        # For training with p3dn instance use - ml.p3dn.24xlarge, with p4dn instance use - ml.p4d.24xlarge\n",
@@ -170,4 +170,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}