From e802a4f0ae58dfbf8277eaa87a9ad10bd7666447 Mon Sep 17 00:00:00 2001 From: yelias Date: Wed, 25 Sep 2024 22:24:48 +0300 Subject: [PATCH] Small fix Signed-off-by: yelias --- examples/tensorflow/dist-mnist/Dockerfile | 3 ++- examples/tensorflow/mnist_utils.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/examples/tensorflow/dist-mnist/Dockerfile b/examples/tensorflow/dist-mnist/Dockerfile index b0d8fc7d86..e7fd5b0563 100644 --- 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 index 8a698eeb82..8cecd3f462 100644 --- 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.