diff --git a/benchmark/paddle/image/googlenet.py b/benchmark/paddle/image/googlenet.py index d3dc0506d58dc7cd2db7b8c3de47f9d0900b3e77..7059c13bd2c2b98eb3fbcf633a6f7064e54d5402 100644 --- a/benchmark/paddle/image/googlenet.py +++ b/benchmark/paddle/image/googlenet.py @@ -16,7 +16,11 @@ args = { 'is_infer': is_infer } define_py_data_sources2( - "train.list", "test.list", module="provider", obj="process", args=args) + "train.list" if not is_infer else None, + "test.list" if is_infer else None, + module="provider", + obj="process", + args=args) settings( batch_size=batch_size, diff --git a/benchmark/paddle/image/resnet.py b/benchmark/paddle/image/resnet.py index 163394e5660dac43b77f125da8b023968d6bc28a..4a14363ff1db48a5072cbb5f5eb3bc9241ffca8f 100644 --- a/benchmark/paddle/image/resnet.py +++ b/benchmark/paddle/image/resnet.py @@ -16,7 +16,11 @@ args = { 'is_infer': is_infer } define_py_data_sources2( - "train.list", "test.list", module="provider", obj="process", args=args) + "train.list" if not is_infer else None, + "test.list" if is_infer else None, + module="provider", + obj="process", + args=args) settings( batch_size=batch_size, diff --git a/benchmark/paddle/image/vgg.py b/benchmark/paddle/image/vgg.py index 2d8075bcf2285253a68cafa778d9bff114cd4c32..8d0a1e97a451cd52ef17e4e326673cc90059ef3c 100644 --- a/benchmark/paddle/image/vgg.py +++ b/benchmark/paddle/image/vgg.py @@ -16,7 +16,11 @@ args = { 'is_infer': is_infer } define_py_data_sources2( - "train.list", "test.list", module="provider", obj="process", args=args) + "train.list" if not is_infer else None, + "test.list" if is_infer else None, + module="provider", + obj="process", + args=args) settings( batch_size=batch_size,