Skip to content

Add DLRM Model#1171

Merged
marcromeyn merged 4 commits into
NVIDIA-Merlin:mainfrom
edknv:torch/dlrm_model
Jul 3, 2023
Merged

Add DLRM Model#1171
marcromeyn merged 4 commits into
NVIDIA-Merlin:mainfrom
edknv:torch/dlrm_model

Conversation

@edknv
Copy link
Copy Markdown
Contributor

@edknv edknv commented Jul 3, 2023

This PR adds a DLRMModel function that builds a Model using the DLRMBlock added in #1162.

@edknv edknv added enhancement New feature or request area/pytorch labels Jul 3, 2023
@github-actions
Copy link
Copy Markdown

github-actions Bot commented Jul 3, 2023

Documentation preview

https://nvidia-merlin.github.io/models/review/pr-1171

Comment thread merlin/models/torch/models/ranking.py Outdated
from merlin.schema import Schema


def DLRMModel(
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

I think it would be better to turn this into a class that extends Model. That would provide more flexibility to our users since they could extend it. In the past, I think we might have over-used using functions (would make sense if the output-type could differ).

@edknv edknv marked this pull request as ready for review July 3, 2023 11:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area/pytorch enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants