From a14bb9580f3442ca0b39efcd13707cc6f2c424b2 Mon Sep 17 00:00:00 2001 From: Marek Drozdowski Date: Mon, 17 Dec 2018 19:51:48 -0800 Subject: [PATCH] fixed validation files requirement for synthetic --- tftrt/examples/image-classification/image_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tftrt/examples/image-classification/image_classification.py b/tftrt/examples/image-classification/image_classification.py index 0090b802a..06cc1e468 100644 --- a/tftrt/examples/image-classification/image_classification.py +++ b/tftrt/examples/image-classification/image_classification.py @@ -553,7 +553,7 @@ def get_frozen_graph( if args.precision != 'fp32' and not args.use_trt: raise ValueError('TensorRT must be enabled for fp16 or int8 modes (--use_trt).') - if args.precision == 'int8' and not args.calib_data_dir: + if args.precision == 'int8' and not args.calib_data_dir and not args.use_synthetic: raise ValueError('--calib_data_dir is required for int8 mode') if args.num_iterations is not None and args.num_iterations <= args.num_warmup_iterations: raise ValueError('--num_iterations must be larger than --num_warmup_iterations '