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.02.2
Copied!
| Name | Size | Updated | Actions |
|---|---|---|---|
dataloader.py | 3.88 KB | August 30, 2021 UTC | |
distributed_utils.py | 4.28 KB | August 30, 2021 UTC | |
Dockerfile | 1.19 KB | August 30, 2021 UTC | |
Dockerfile_spark | 1.88 KB | August 30, 2021 UTC | |
embedding.py | 4.14 KB | August 30, 2021 UTC | |
interaction.py | 2.62 KB | August 30, 2021 UTC | |
lr_scheduler.py | 2.53 KB | August 30, 2021 UTC | |
main.py | 14.66 KB | August 30, 2021 UTC | |
model.py | 25.69 KB | August 30, 2021 UTC | |
README.md | 37.7 KB | August 30, 2021 UTC | |
requirements.txt | 100 B | August 30, 2021 UTC | |
slurm_multinode.sh | 1.55 KB | August 30, 2021 UTC | |
split_binary_dataset.py | 9.47 KB | August 30, 2021 UTC |