sxjscience opened a new issue #18617: URL: https://github.com/apache/incubator-mxnet/issues/18617
For example, some losses won't support setting `batch_axis`, e.g.: https://github.com/apache/incubator-mxnet/blob/4b86c32832a994e76b97dfc58c8a672db87e721d/python/mxnet/gluon/loss.py#L299-L302 Also, there are cases like HuberLoss that hasn’t supported numpy: https://github.com/apache/incubator-mxnet/blob/4b86c32832a994e76b97dfc58c8a672db87e721d/python/mxnet/gluon/loss.py#L604-L610 For the HuberLoss, it is tracked in https://github.com/apache/incubator-mxnet/pull/18603. However, there are more loss functions that do not support numpy. @yzhliu @szha @leezu I mark this as 2.0 because we may keep the loss layers. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org