Merge pull request #3383 from chiamp:imagenet

PiperOrigin-RevId: 571958738

examples/imagenet/README.md

@@ -23,6 +23,7 @@ Table of contents:
 - [Example runs](#example-runs)
 - [Running locally](#running-locally)
   - [Overriding parameters on the command line](#overriding-parameters-on-the-command-line)
+- [Running fake data benchmarks](#running-fake-data-benchmarks)
 - [Running on Cloud](#running-on-cloud)
   - [Preparing the dataset](#preparing-the-dataset)
   - [Google Cloud TPU](#google-cloud-tpu)
@@ -66,6 +67,22 @@ python main.py --workdir=./imagenet_default --config=configs/default.py \
 --config.num_epochs=100
 ```
 
+### Running fake data benchmarks
+
+Execute the following code with `flax/examples/imagenet` as your current directory:
+```shell
+bash ../../tests/download_dataset_metadata.sh
+python imagenet_fake_data_benchmark.py
+```
+
+If you get an error like this:
+```shell
+Cloning into 'datasets'...
+fatal: cannot change to 'https://github.com/tensorflow/datasets/': No such file or directory
+error: failed to initialize sparse-checkout
+```
+This mean your git version is outdated. Just update it and re-run.
+
 ### Running on Cloud
 
 #### Preparing the dataset