Ian commented on issue #14333: Predict of mx.mod.Module api problem
URL:
https://github.com/apache/incubator-mxnet/issues/14333#issuecomment-470824608
> Yes, it's normal. You should add a argmax layer on the top of the network
to get the label
Thanks, your suggest. I can get lab
Ian commented on issue #14333: Predict of mx.mod.Module api problem
URL:
https://github.com/apache/incubator-mxnet/issues/14333#issuecomment-470396433
> @Ian Can you shed some light on the model you are using ?
> Specifically, what's the last layer of your model ? Is it Sigmoid o