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.1
Copied!
| Name | Size | Updated | Actions |
|---|---|---|---|
dataloader.py | 4.42 KB | February 3, 2022 UTC | |
distributed_utils.py | 4.28 KB | February 3, 2022 UTC | |
Dockerfile | 1.18 KB | February 3, 2022 UTC | |
Dockerfile_spark | 1.92 KB | February 3, 2022 UTC | |
embedding.py | 2.02 KB | February 3, 2022 UTC | |
interaction.py | 2.62 KB | February 3, 2022 UTC | |
lr_scheduler.py | 2.53 KB | February 3, 2022 UTC | |
main.py | 16.54 KB | February 3, 2022 UTC | |
model.py | 28.5 KB | February 3, 2022 UTC | |
README.md | 37.7 KB | February 3, 2022 UTC | |
requirements.txt | 100 B | February 3, 2022 UTC | |
slurm_multinode.sh | 1.55 KB | February 3, 2022 UTC | |
split_binary_dataset.py | 9.47 KB | February 3, 2022 UTC |