Is conversion from a coreml multi-output model supported?

Hi,
I was wondering if conversion from a coreml multi-output model is support? I have a coreml model with multiple outputs, and each output has different output shapes. I checked several threads, and it seems that we can use get_output(i). But my get_num_outputs() only gave me 1, so the compiled nnvm model only has 1 output. I was wondering whether the conversion is not supported yet or I should add other parameters during conversion. Thanks.

CorelML frontend is returning the first output. You can try editing here to return multiple outputs

1 Like

I edited the code to return multiple outputs (https://github.com/jjiang2cal/tvm/commit/6c384eaa7b57d9b2e5656967b90d7ee693772a2a). It worked on my custom resnet model. Will consider submit a PR after adding tests.

1 Like