Merge pull request #4507 from guoshengCS/fix-infer-none

Fix infer when input is empty in v2/inference.py
tonyyang-svail-feed-op-desgin
Guo Sheng 8 years ago committed by GitHub
commit cf928020db

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