Model
Wide&Deep Base TensorFlow2 checkpoint trained with AMP on NVTabular preprocessed dataset with multihot embeddings
Use the NGC CLI to download:
Copied!
1 Version
22.03.0_ampSelected09/22/2022 5:02 PM UTC702.85 MBAccuracy: 020 EpochsBatch Size: 131072GPU: V100 Copied!
22.03.0_ampSelected
09/22/2022 5:02 PM UTC702.85 MBAccuracy: 020 EpochsBatch Size: 131072GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| type | Base |
performance
| Key | Value |
|---|---|
| MAP@12 | 0.6575 |
training
| Key | Value |
|---|---|
| batch_size | 131072 |
| lr_deep | 0.00012 |
| lr_wide | 0.02 |
| training_precision | AMP |
| dataset | Outbrain |
| epochs | 20 |
| combiner | sum |