Coremltools: Specify output tensor names for PyTorch converted model
鉂換uestion
Hi,
I am exporting a model from PyTorch with multiple output tensors, but those tensors in the exported model have numerical meaningless names.
Any chance I can specify the output names just like it is done with the input name?
Roulbac
>All comments
While we should be able to specify "output" in the convert function for a PyTorch model, and this has been requested a few times like in #775, you might want to look at the "utils" https://coremltools.readme.io/docs/mlmodel-utilities which allow renaming of the output.
leovinus2001
on 22 Aug 2020
👍3
Was this page helpful?
0 / 5 - 0 ratings
Related issues
baaj2109
路
4Comments
ghost
路
3Comments
tucan9389
路
4Comments
mushipand
路
5Comments
xternalz
路
3Comments
Most helpful comment
While we should be able to specify "output" in the convert function for a PyTorch model, and this has been requested a few times like in #775, you might want to look at the "utils" https://coremltools.readme.io/docs/mlmodel-utilities which allow renaming of the output.