for future reference and people like myself :slight_smile:
the last layer doesn't need to be 1 output for softmax cross entropy to work:
nn.Dense(10)
this one was enough to make the loss function work properly .
---
[Visit
Topic](https://discuss.mxnet.apache.org/t/help-with-simple-classification/6586/4)
or reply to this email to respond.
You are receiving this because you enabled mailing list mode.
To unsubscribe from these emails, [click
here](https://discuss.mxnet.apache.org/email/unsubscribe/bfb9a350351fda04f2ac5dda2c5e9a601c592b7273b31fcfcb36b73df233c313).
