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

Refine embedding table info #1768

Merged
merged 6 commits into from Feb 21, 2020
Merged

Conversation

QiJune
Copy link
Collaborator

@QiJune QiJune commented Feb 21, 2020

This PR does two things:

  • rename EmbeddingTableInfo to EmbeddingTableDesc in Model messge
  • set dtype to float32 when worker pushing embedding table desc. Otherwise, GO PS could not create an embedding table correctly.

Copy link
Contributor

@lwpyr lwpyr left a comment

Choose a reason for hiding this comment

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

It looks like some modification over names, approved.

@@ -49,7 +49,7 @@ func (model *Model) SetEmbeddingTableInfo(info *proto.EmbeddingTableInfo) {

// InitFromModelPB inits the model from model PB
func (model *Model) InitFromModelPB(pb *proto.Model) error {
for _, v := range pb.EmbeddingTableInfo {
for _, v := range pb.EmbeddingTableInfos {
Copy link
Collaborator

Choose a reason for hiding this comment

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

Info has no plural form.

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

How about change Info to Desc? Desc is short for Description

Copy link
Collaborator

Choose a reason for hiding this comment

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

descriptions/attributes/properties/information. Hard to choose a better one.

We can keep to use EmbeddingTableInfos, since it can be understood.

@QiJune QiJune merged commit 405c6fa into sql-machine-learning:develop Feb 21, 2020
@QiJune QiJune deleted the clean/proto branch August 7, 2020 02:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants