I used mlmodel from PyTorch- convert PyTorch to onnx and then convert it to mlmodel.
I don’t want to do the task of softmax layer in the field of mobile, so I hope let Machine learning engineer to change the code to do the softmax layer .
For instance how to change the input and output type.I did image classifier .