BentoML icon indicating copy to clipboard operation
BentoML copied to clipboard

Published 20 hours ago • verified publisher icon bentoml

ONNX Runner Only return first of the outputs!

Open cristozz opened this issue 3 years ago • 1 comments

https://github.com/bentoml/BentoML/blob/28f0bc320637851a158cafaed90ef15bb87fe82a/bentoml/_internal/frameworks/onnx.py#L415

I have a onnx model that have two outputs ['label', 'probabilities'] but when i use RUN only return 'label' values, and i dont see any configuration to return all outputs.

cristozz avatar Sep 16 '22 00:09 cristozz

Hi @cristozz, I believe this issue is addressed in https://github.com/bentoml/BentoML/pull/2912, it will be available in the upcoming release version 1.1.

@larme could you confirm if this is the same issue that #2912 fixed?

parano avatar Sep 16 '22 18:09 parano

Hi guys, I am wondering whether we have any progress on this issue. Thanks

petcai avatar Oct 07 '22 08:10 petcai

Hi @petcai sorry about the delay in getting back to you here, this issue has been fixed in #2912 and released since 1.0.7

parano avatar Oct 31 '23 17:10 parano