Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[Numpy][Gluon] Multiple gluon loss functions do not support numpy #18617

Open
sxjscience opened this issue Jun 24, 2020 · 2 comments
Open

[Numpy][Gluon] Multiple gluon loss functions do not support numpy #18617

sxjscience opened this issue Jun 24, 2020 · 2 comments

Comments

@sxjscience
Copy link
Member

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 #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.

@szha
Copy link
Member

szha commented Jun 24, 2020

I started working on this in #17196 but didn't get the time to finish yet. I will try to prioritize.

@szha
Copy link
Member

szha commented Jul 26, 2020

The PR should be ready now.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

No branches or pull requests

2 participants