This repository has been archived by the owner on Nov 17, 2023. It is now read-only.
Replies: 2 comments
-
Hey, this is the MXNet Label Bot. |
Beta Was this translation helpful? Give feedback.
0 replies
-
@mxnet-label-bot add [operator, question] |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Description
The
take
operator currently does not supportgrad_stype='row_sparse'
, which can impact performance. This post describes the issue and a possible work-around viaEmbedding
. However, full support is still desirable, becausetake
is a more general operator thanEmbedding
.Environment info (Required)
Package used (Python/R/Scala/Julia): Python
Error Message:
Minimum reproducible example
This is the original approach that triggers the warning
This is the work-around that preserves
grad_stype='row_sparse'
:What have you tried to solve it?
Use the alternative example in case row_sparse gradients are desired.
Beta Was this translation helpful? Give feedback.
All reactions