Model
Wide & Deep Base TensorFlow checkpoint trained with AMP
Use the NGC CLI to download:
Copied!
1 Version
20.06.0_ampSelected09/22/2022 4:47 PM UTC769.74 MBAccuracy: 0.816120 EpochsBatch Size: 131072GPU: V100 Copied!
20.06.0_ampSelected
09/22/2022 4:47 PM UTC769.74 MBAccuracy: 0.816120 EpochsBatch Size: 131072GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| layers | 1024 1024 1024 1024 1024 |
| type | base |
performance
| Key | Value |
|---|---|
| map_with_leak | 0.677 |
| accuracy | 0.816 |
training
| Key | Value |
|---|---|
| batch_size | 131072 |
| LR_wide | 0.2 |
| LR_deep | 1 |
| training_precision | AMP |
| epochs | 120 |
| dataset | Outbrain |
| iterations | 54713 |