TF Example: specify steps arg to tf.Estimator.evaluate for ending the… (#293)

* TF Example: specify steps arg to tf.Estimator.evaluate for ending the Evaluation

Signed-off-by: Serge Panev <spanev@nvidia.com>

* TF Example: add log dir for evaluation

Signed-off-by: Serge Panev <spanev@nvidia.com>

Author: Kh4L

docs/examples/tensorflow/demo/nvutils/runner.py

@@ -305,6 +305,7 @@ def validate(infer_func, params):
 
     filename_pattern = os.path.join(data_dir, '%s-*')
     eval_filenames  = sorted(tf.gfile.Glob(filename_pattern % 'validation'))
+    num_eval_samples = _get_num_records(eval_filenames)
 
     eval_idx_filenames = None
     if data_idx_dir is not None:
@@ -362,7 +363,8 @@ def validate(infer_func, params):
                     training=False, distort_color=False,
                     deterministic=deterministic,
                     dali_cpu=dali_cpu, idx_filenames=eval_idx_filenames,
-                    num_threads=num_preproc_threads))
+                    num_threads=num_preproc_threads),
+                    steps=num_eval_samples)
             print('Top-1 accuracy:', eval_result['top1_accuracy']*100, '%')
             print('Top-5 accuracy:', eval_result['top5_accuracy']*100, '%')
         except KeyboardInterrupt:

qa/L1_tensorflow-dali_test/test.sh

@@ -56,7 +56,7 @@ test_body() {
         python -u resnet.py --layers=18 \
         --data_dir=/data/imagenet/train-val-tfrecord-480-subset --data_idx_dir=idx-files/ \
         --precision=fp16 --num_iter=100  --iter_unit=batch --display_every=50 \
-        --batch=256 --dali_cpu
+        --batch=256 --dali_cpu --log_dir=dali_log
 }
 
 source ../../../../qa/test_template.sh