Taran Iyengar

DLRM v2 Preprocessed Multihot Criteo day_23 Dataset Accuracy Drop

18

From the step at https://github.com/mlcommons/training/blob/3a6a379305e6ef0a1c34461c823e9c99d52c1021/recommendation_v2/torchrec_dlrm/scripts/process_Criteo_1TB_Click_Logs_dataset.sh#L42, there was a difference with the generated day_23_sparse.py using torchrec==0.3.2 resulting in an roc_auc of 61.64% expected row_0 array([[ 10540786, 197, 34, ..., 34, 2,...

Fused Neigbhor Sampling reproducibility issue

3

Setting the seed and repeating the fused neighborhood sampling for a source code does not reproduce the same subgraph, have identified a fix that will be slower but allow reproducible...