Fix infer when input is empty in v2/inference.py

tonyyang-svail-feed-op-desgin
guosheng 8 years ago
parent e82fda2521
commit 46641c6361

@ -96,6 +96,9 @@ class Inference(object):
for i, item in enumerate(result):
retv[i].append(item)
if retv == None:
return []
if flatten_result:
retv = [numpy.concatenate(out) for out in retv]

Loading…
Cancel
Save