提交 d14ca1c3 编写于 作者: T tensor-tang

fix inference crash of alexnet benchmark

上级 5bf37b91
...@@ -19,7 +19,11 @@ args = { ...@@ -19,7 +19,11 @@ args = {
'num_samples': num_samples 'num_samples': num_samples
} }
define_py_data_sources2( define_py_data_sources2(
"train.list", None, 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( settings(
batch_size=batch_size, batch_size=batch_size,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册