mobile-semantic-segmentation icon indicating copy to clipboard operation
mobile-semantic-segmentation copied to clipboard
verified publisher icon akirasosa

Output data type is MultiArray after run_convert_coreml.py

Open maromaSamsa opened this issue 3 years ago • 0 comments

Hi, after training is completed, I want the model to convert to .mlmodel format, but I got this kind of file below which output is not same as input data type:

Image

Am I miss something? I have search coremltools API ref, it seems there has no way to config output in pytorch, does someone know how I can fix it?

maromaSamsa avatar Feb 21 '22 08:02 maromaSamsa