Merge pull request #7115 from tensor-tang/alexnet

fix inference crash of alexnet benchmark
del_some_in_makelist
Tao Luo 7 years ago committed by GitHub
commit 7438bd98ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -19,7 +19,11 @@ args = {
'num_samples': num_samples
}
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(
batch_size=batch_size,

Loading…
Cancel
Save