huifengguo opened a new issue #17728: Poor performance when we use symbol.(..,..,stype="row-sparse") to define the variable URL: https://github.com/apache/incubator-mxnet/issues/17728 ## Description Task: CTR prediction Model: DCN Dataset: Criteo Everything is same except the definition of embedding variable: ``` // default embed_weight = mx.symbol.Variable('embed_weight', stype='default') // row_sparse embed_weight = mx.symbol.Variable('embed_weight', stype='row_sparse') ``` ` embed = mx.symbol.Embedding(data=ids, weight=embed_weight, input_dim=200000, output_dim=hidden_units[0], sparse_grad=True) ` ## Occurrences When stype="default", the result of AUC is 0.802, while the test result of AUC is 0.794 when stype="row-sparse". ## What have you tried to solve it? 1. 2.
---------------------------------------------------------------- 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 With regards, Apache Git Services
