NVIDIA Deep Learning Examples
DLRM TensorFlow2 codebase
Resource
NVIDIA Deep Learning Examples
DLRM TensorFlow2 codebase

TensorFlow2 codebase for training and using DLRM model

NameSizeUpdatedActions
dataloader.py
4.69 KBNovember 4, 2022 UTC
datasets.py
9.15 KBNovember 4, 2022 UTC
defaults.py
844 BNovember 4, 2022 UTC
Dockerfile
1.69 KBNovember 4, 2022 UTC
Dockerfile_spark
1.92 KBNovember 4, 2022 UTC
embedding.py
11.48 KBNovember 4, 2022 UTC
feature_spec.py
14.23 KBNovember 4, 2022 UTC
gen_csv.py
2.74 KBNovember 4, 2022 UTC
interaction.py
2.62 KBNovember 4, 2022 UTC
lr_scheduler.py
2.53 KBNovember 4, 2022 UTC
main.py
16.49 KBNovember 4, 2022 UTC
model.py
22.05 KBNovember 4, 2022 UTC
prepare_synthetic_dataset.py
6.16 KBNovember 4, 2022 UTC
README.md
57.05 KBNovember 4, 2022 UTC
requirements.txt
107 BNovember 4, 2022 UTC
slurm_multinode.sh
1.46 KBNovember 4, 2022 UTC