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.11.0
Copied!
| Name | Size | Updated | Actions |
|---|---|---|---|
dataloader.py | 4.63 KB | July 8, 2022 UTC | |
datasets.py | 8.57 KB | July 8, 2022 UTC | |
defaults.py | 844 B | July 8, 2022 UTC | |
distributed_utils.py | 6.13 KB | July 8, 2022 UTC | |
Dockerfile | 1.18 KB | July 8, 2022 UTC | |
Dockerfile_spark | 1.92 KB | July 8, 2022 UTC | |
embedding.py | 13.4 KB | July 8, 2022 UTC | |
feature_spec.py | 14.23 KB | July 8, 2022 UTC | |
gen_csv.py | 2.74 KB | July 8, 2022 UTC | |
interaction.py | 2.62 KB | July 8, 2022 UTC | |
lr_scheduler.py | 2.53 KB | July 8, 2022 UTC | |
main.py | 17.47 KB | July 8, 2022 UTC | |
model.py | 32.42 KB | July 8, 2022 UTC | |
prepare_synthetic_dataset.py | 6.16 KB | July 8, 2022 UTC | |
README.md | 58.38 KB | July 8, 2022 UTC | |
requirements.txt | 107 B | July 8, 2022 UTC | |
slurm_multinode.sh | 1.54 KB | July 8, 2022 UTC |