skip train list when inference, skip test list when training

release/0.11.0
tensor-tang 7 years ago
parent 06d24e3a85
commit aef639448c

@ -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,

@ -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,

@ -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,

Loading…
Cancel
Save