Resource
NVIDIA Deep Learning Examples
DLRM for TensorFlow2The Deep Learning Recommendation Model (DLRM) is a recommendation model designed to make use of both categorical and numerical inputs.
21.06.0
Copied!
| Name | Size | Updated | Actions |
|---|---|---|---|
dataloader.py | 4.42 KB | November 18, 2021 UTC | |
distributed_utils.py | 4.28 KB | November 18, 2021 UTC | |
Dockerfile | 1.19 KB | November 18, 2021 UTC | |
Dockerfile_spark | 1.88 KB | November 18, 2021 UTC | |
embedding.py | 2.02 KB | November 18, 2021 UTC | |
interaction.py | 2.62 KB | November 18, 2021 UTC | |
lr_scheduler.py | 2.53 KB | November 18, 2021 UTC | |
main.py | 16.33 KB | November 18, 2021 UTC | |
model.py | 27.07 KB | November 18, 2021 UTC | |
README.md | 37.7 KB | November 18, 2021 UTC | |
requirements.txt | 100 B | November 18, 2021 UTC | |
slurm_multinode.sh | 1.55 KB | November 18, 2021 UTC | |
split_binary_dataset.py | 9.47 KB | November 18, 2021 UTC |