decrease version sm tf training

Author: Jyothirmaikottu

tensorflow/training/docker/2.18/py3/Dockerfile.cpu

@@ -283,7 +283,7 @@ RUN $PYTHON -m pip install --no-cache-dir -U \
     sagemaker-training==4.8.3
 
     RUN $PYTHON -m pip install --no-cache-dir -U \
-    sagemaker-tensorflow-training==20.5.0
+    sagemaker-tensorflow-training==20.4.1
 
 RUN $PYTHON -m pip install --no-cache-dir -U \
     sagemaker-studio-analytics-extension==0.1.4

tensorflow/training/docker/2.18/py3/Dockerfile.sagemaker.cpu.core_packages.json

@@ -15,7 +15,7 @@
     "version_specifier": ">=2.207.1,<3"
   },
   "sagemaker-tensorflow-training": {
-    "version_specifier": "==20.5.0"
+    "version_specifier": "==20.4.1"
   },
   "sagemaker-training": {
     "version_specifier": "==4.8.3"

tensorflow/training/docker/2.18/py3/cu125/Dockerfile.gpu

@@ -386,7 +386,7 @@ RUN $PYTHON -m pip install --no-cache-dir -U \
     sagemaker-training==4.8.3
 
     RUN $PYTHON -m pip install --no-cache-dir -U \
-    sagemaker-tensorflow-training==20.5.0
+    sagemaker-tensorflow-training==20.4.1
 
 RUN $PYTHON -m pip install --no-cache-dir -U \
     sagemaker-studio-analytics-extension==0.1.4

tensorflow/training/docker/2.18/py3/cu125/Dockerfile.sagemaker.gpu.core_packages.json

@@ -15,7 +15,7 @@
     "version_specifier": ">=2.207.1,<3"
   },
   "sagemaker-tensorflow-training": {
-    "version_specifier": "==20.5.0"
+    "version_specifier": "==20.4.1"
   },
   "sagemaker-training": {
     "version_specifier": "==4.8.3"