Model
DLRM PyTorch checkpoint trained on Criteo Dataset with FreqLimit=15 on A100 without AMP
Use the NGC CLI to download:
Copied!
2 Versions
22.06.0_fp32Selected09/22/2022 6:06 PM UTC15.61 GBAccuracy: 0.802581 EpochsBatch Size: 65536GPU: A100 Copied!
22.06.0_fp32Selected
09/22/2022 6:06 PM UTC15.61 GBAccuracy: 0.802581 EpochsBatch Size: 65536GPU: A100
Copied!
architecture
| Key | Value |
|---|---|
| type | base |
performance
| Key | Value |
|---|---|
| accuracy | 0.80258 |
| training_loss | 0.12573 |
| AUC | 0.80258 |
training
| Key | Value |
|---|---|
| batch_size | 65536 |
| training_precision | AMP |
| dataset | Criteo (FL15) |
| epochs | 1 |
09/22/2022 5:57 PM UTC15.61 GBAccuracy: 0.802531 EpochsBatch Size: 65536GPU: A100 Copied!
09/22/2022 5:57 PM UTC15.61 GBAccuracy: 0.802531 EpochsBatch Size: 65536GPU: A100
Copied!
architecture
| Key | Value |
|---|---|
| type | base |
performance
| Key | Value |
|---|---|
| accuracy | 0.80253 |
| training_loss | 0.12574 |
| AUC | 0.80253 |
training
| Key | Value |
|---|---|
| batch_size | 65536 |
| training_precision | AMP |
| dataset | Criteo (FL15) |
| epochs | 1 |