From aef639448c67999e3bfc094c6d39ca528fe193a4 Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Thu, 30 Nov 2017 14:33:43 +0800 Subject: [PATCH] skip train list when inference, skip test list when training --- benchmark/paddle/image/googlenet.py | 6 +++++- benchmark/paddle/image/resnet.py | 6 +++++- benchmark/paddle/image/vgg.py | 6 +++++- 3 files changed, 15 insertions(+), 3 deletions(-) diff --git a/benchmark/paddle/image/googlenet.py b/benchmark/paddle/image/googlenet.py index d3dc0506d5..7059c13bd2 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 163394e566..4a14363ff1 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 2d8075bcf2..8d0a1e97a4 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, -- GitLab