xinyu-intel commented on issue #11086: Infer shape error with softmaxoutput
URL:
https://github.com/apache/incubator-mxnet/issues/11086#issuecomment-393192905
Yes, you are right. By the way, is it possible to automatically perform
flatten before softmax when the softmax input is such (N,
xinyu-intel commented on issue #11086: Infer shape error with softmaxoutput
URL:
https://github.com/apache/incubator-mxnet/issues/11086#issuecomment-393038694
This change aims to align Caffe's
[MobileNet](https://github.com/shicai/MobileNet-Caffe/blob/master/mobilenet_deploy.prototxt)
xinyu-intel commented on issue #11086: Infer shape error with softmaxoutput
URL:
https://github.com/apache/incubator-mxnet/issues/11086#issuecomment-392758330
Temporarily add flatten manually before softmax to avoid this problem.