Skip to content

support sparse output for lookup table grad op#5145

Merged
QiJune merged 29 commits intoPaddlePaddle:developfrom
QiJune:lookup_table_op_sparse
Oct 28, 2017
Merged

support sparse output for lookup table grad op#5145
QiJune merged 29 commits intoPaddlePaddle:developfrom
QiJune:lookup_table_op_sparse

Conversation

@QiJune
Copy link
Member

@QiJune QiJune commented Oct 27, 2017

Add is_sparse attribute for lookup table operator. And book04.word2vec can run in sparse update mode. But there is some memory error after some batches training. It can be generated stably after 21 batches. So, I will debug this to make sure sparse update is worked.

Updated:
The bug has been found with the help of @reyoung . In Lookup Table Grad operator, there is a bug of array bounds.

reyoung
reyoung previously approved these changes Oct 27, 2017
Copy link
Collaborator

@reyoung reyoung left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@QiJune QiJune merged commit 008f40c into PaddlePaddle:develop Oct 28, 2017
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

2 participants