Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

cpu 分布式训练 多值特征应该怎么输入到sparse_embedding #68431

Open
welsonzhang opened this issue Sep 25, 2024 · 0 comments
Open
Assignees
Labels

Comments

@welsonzhang
Copy link

请提出你的问题 Please ask your question

#38721 看了这问题,并没有给出解决方案,我也遇到类似的问题。

cpu分布式训练的时候,有个slot下面会有多个feasign,但是sparse_embedding,最后一个维度只能是1,因此将多个feasign扩展了1的维度,变成[batch_size, slot_num, 1], 但是sparse_embedding的api提示:
[Hint: Expected ids_dim.size() == 2, but received ids_dim.size():3 != 2:2.] (at /paddle/paddle/fluid/operators/pscore/distributed_lookup_table_op.cc:56)

为什么会有这个限制,如果是多值特征,应该怎么处理?

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

No branches or pull requests

2 participants