DLRM OneEmbedding Graph Train and Eval - Githubissues

Oneflow-Inc / models

Models and examples built with OneFlow

Apache License 2.0

94 stars 37 forks source link

DLRM OneEmbedding Graph Train and Eval #323

Closed ShawnXuan closed 2 years ago

ShawnXuan commented 2 years ago

OneFlow DLRM model with OneEmbedding train and evaluation in graph mode.

[x] README
[x] parquet dataset script