@@ -194,7 +194,7 @@ To train your model using mixed precision or TF32 with Tensor Cores or FP32, per
1941941 . Clone the repository.
195195```
196196git clone https://github.com/NVIDIA/DeepLearningExamples
197- cd DeepLearningExamples/TensorFlow/Classification/RN50v1.5
197+ cd DeepLearningExamples/TensorFlow/Classification/ConvNets
198198```
199199
2002002 . Download and preprocess the dataset.
@@ -452,10 +452,9 @@ To benchmark the training performance on a specific batch size, run:
452452Each of these scripts runs 200 warm-up iterations and measures the first epoch.
453453
454454To control warmup and benchmark length, use the ` --warmup_steps ` , ` --num_iter ` and ` --iter_unit ` flags. Features like XLA or DALI can be controlled
455- with ` --use_xla ` and ` --use_dali ` flags.
455+ with ` --use_xla ` and ` --use_dali ` flags. If no ` --data_dir=<path to imagenet> ` flag is specified then the benchmarks will use a synthetic dataset.
456456Suggested batch sizes for training are 256 for mixed precision training and 128 for single precision training per single V100 16 GB.
457457
458-
459458#### Inference performance benchmark
460459
461460To benchmark the inference performance on a specific batch size, run:
@@ -470,6 +469,7 @@ To benchmark the inference performance on a specific batch size, run:
470469
471470By default, each of these scripts runs 20 warm-up iterations and measures the next 80 iterations.
472471To control warm-up and benchmark length, use the ` --warmup_steps ` , ` --num_iter ` and ` --iter_unit ` flags.
472+ If no ` --data_dir=<path to imagenet> ` flag is specified then the benchmarks will use a synthetic dataset.
473473
474474The benchmark can be automated with the ` inference_benchmark.sh ` script provided in ` resnet50v1.5 ` , by simply running:
475475` bash ./resnet50v1.5/inference_benchmark.sh <data dir> <data idx dir> `
0 commit comments