Skip to content

Commit aef6394

Browse files
committed
skip train list when inference, skip test list when training
1 parent 06d24e3 commit aef6394

File tree

3 files changed

+15
-3
lines changed

3 files changed

+15
-3
lines changed

benchmark/paddle/image/googlenet.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,11 @@
1616
'is_infer': is_infer
1717
}
1818
define_py_data_sources2(
19-
"train.list", "test.list", module="provider", obj="process", args=args)
19+
"train.list" if not is_infer else None,
20+
"test.list" if is_infer else None,
21+
module="provider",
22+
obj="process",
23+
args=args)
2024

2125
settings(
2226
batch_size=batch_size,

benchmark/paddle/image/resnet.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,11 @@
1616
'is_infer': is_infer
1717
}
1818
define_py_data_sources2(
19-
"train.list", "test.list", module="provider", obj="process", args=args)
19+
"train.list" if not is_infer else None,
20+
"test.list" if is_infer else None,
21+
module="provider",
22+
obj="process",
23+
args=args)
2024

2125
settings(
2226
batch_size=batch_size,

benchmark/paddle/image/vgg.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,11 @@
1616
'is_infer': is_infer
1717
}
1818
define_py_data_sources2(
19-
"train.list", "test.list", module="provider", obj="process", args=args)
19+
"train.list" if not is_infer else None,
20+
"test.list" if is_infer else None,
21+
module="provider",
22+
obj="process",
23+
args=args)
2024

2125
settings(
2226
batch_size=batch_size,

0 commit comments

Comments
 (0)