Small fix

Signed-off-by: yelias <[email protected]>
YosiElias · Sep 25, 2024 · e802a4f · e802a4f
1 parent 4777e29
commit e802a4f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/examples/tensorflow/dist-mnist/Dockerfile b/examples/tensorflow/dist-mnist/Dockerfile
@@ -14,7 +14,8 @@
 
 FROM tensorflow/tensorflow:2.17.0
 
-# Using keras-2.17 because of bug on keras-3.4.1 which used by default by TF-2.17
+# Using keras-2.17 because of bug on keras-3.4.1
+# which used by default by TF-2.17 (https://github.com/tensorflow/tensorflow/issues/72388)
 ENV TF_USE_LEGACY_KERAS 1
 RUN pip install tf_keras
 

diff --git a/examples/tensorflow/mnist_utils.py b/examples/tensorflow/mnist_utils.py
@@ -29,7 +29,7 @@ def load_data(fake_data=False, data_path=None, repeat=False):
      Loads the MNIST dataset and converts it into TensorFlow datasets.
 
     Args:
-         fake_data (bool): If `True`, loads a fake dataset for testing purposes.
+         fake_data (bool, optional): If `True`, loads a fake dataset for testing purposes.
                            If `False`, loads the real MNIST dataset.
          data_path (str, optional): Path where to cache the dataset locally.
                            If `None`, the dataset is loaded to the default location.