Practical overview
When it fits
Use Dlrm when you need a reference implementation or prototype for a recommendation model that combines dense numerical signals with sparse categorical features and embedding interactions.
Who it is for
- Developers prototyping personalization or recommendation models.
- Researchers and practitioners studying models that combine numerical features with sparse categorical inputs.
Documented uses
- Explore a recommendation model that processes dense and sparse features together.
- Prototype ranking or personalization pipelines using embedding tables and multilayer perceptrons.
- Compare embedding-interaction operators such as sum, dot product and concatenation.
Requirements & limits
The public repository identifies the project as open source under the MIT license.
- The repository includes both PyTorch and Caffe2 implementations.
- The model expects dense floating-point vectors and sparse indices into embedding tables.
- Embedding interactions are described using sum, dot-product or concatenation operators.
- The README provides a tiny PyTorch training example, as well as data-generation/loading code, tests and benchmark scripts.
Check current requirements in the official documentation. Source-based guidance does not mean OpenViST has run or independently tested this project.
Run the DLRM PyTorch tiny-model sample
Train the documented tiny DLRM model using generated data.
Before you start
- Python
- The DLRM repository, including dlrm_s_pytorch.py
Steps
From the repository, run the command below.
python dlrm_s_pytorch.py --mini-batch-size=2 --data-size=6Review the training progress output, which reports iteration, loss, and accuracy.
Expected outcome
The sample reports three training iterations for a tiny model.
Guide limitations
- The README labels this a sample run and does not document environment setup or additional dependency requirements.
- The sample output shows accuracy as 0.000%; it does not establish recommendation quality.
Sources & review
Source record details
- Source availability
- Active
- Sources checked
- 12 October 2026
- Original website title
- facebookresearch/dlrm · Official homepage title · retained in its source language

