Nick Nikolov

Same here.

Thanks! Will use this as reference. Did you model it after the handbook?

Are we sure we not gonna regret this later? can we support both styles by checking?

Try normalising it though

I have the same problem

Seems about right to me (On iPhone 7)

Hey, sorry, the script I was referring to is your Neural Network updatable example. https://coremltools.readme.io/docs/updatable-neural-network-classifier-on-mnist-dataset Basically I want to be able to have a mobilnet-based PyTorch model converted to CoreML...

This is the last 3 layers of my model. As you can see adding the softmax results in type `softmax-nd` Then when calling `builder.set_categorical_cross_entropy_loss` I get the following error ```...

Thanks for following up! Turning off the exceptions works but then I get this warning: ``` python3.10/site-packages/coremltools/models/model.py:145: RuntimeWarning: You will not be able to run predict() on this Core ML...

Hey @TobyRoseman sorry to bother again. Would you mind giving some more guidance? Maybe I'm overlooking some part of the docs I'm not clear on how the SoftmaxND -> Softmax...