NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
DLRM checkpoint (PyTorch, FP32, BS64k, Base, FL15)
Model
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
DLRM checkpoint (PyTorch, FP32, BS64k, Base, FL15)

DLRM PyTorch checkpoint trained on Criteo Dataset with FreqLimit=15 on A100 without AMP

2 Versions
09/22/2022 6:06 PM UTC15.61 GBAccuracy: 0.802581 EpochsBatch Size: 65536GPU: A100
architecture
KeyValue
typebase
performance
KeyValue
accuracy0.80258
training_loss0.12573
AUC0.80258
training
KeyValue
batch_size65536
training_precisionAMP
datasetCriteo (FL15)
epochs1
22.06.0_ampSelected
09/22/2022 5:57 PM UTC15.61 GBAccuracy: 0.802531 EpochsBatch Size: 65536GPU: A100
architecture
KeyValue
typebase
performance
KeyValue
accuracy0.80253
training_loss0.12574
AUC0.80253
training
KeyValue
batch_size65536
training_precisionAMP
datasetCriteo (FL15)
epochs1