Skip to content

DeepFM 1n1d example - #360

Open
MARD1NO wants to merge 3 commits into
mainfrom
deepfm_serving
Open

DeepFM 1n1d example#360
MARD1NO wants to merge 3 commits into
mainfrom
deepfm_serving

Conversation

@MARD1NO

Copy link
Copy Markdown
Contributor

No description provided.

def np_to_global(np):
t = flow.from_numpy(np)
return t.to_global(placement=flow.env.all_device_placement("cpu"), sbp=flow.sbp.split(0))
return t.to_global(

Copy link
Copy Markdown
ContributorAuthor

Choose a reason for hiding this comment

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

单卡情况下,设置sbp为broadcast

Comment threadRecommenderSystems/deepfm/deepfm_train_eval.py
Sign up for freeto 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.

2 participants

@MARD1NO@ShawnXuan