Model
BERT Base PyTorch checkpoint pretrained with LAMB optimizer using AMP
Use the NGC CLI to download:
Copied!
1 Version
19.09.0_ampSelected09/22/2022 5:33 PM UTC1.73 GBAccuracy: 00 EpochsBatch Size: 0GPU: V100 Copied!
19.09.0_ampSelected
09/22/2022 5:33 PM UTC1.73 GBAccuracy: 00 EpochsBatch Size: 0GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| task | Pretraining |
| type | Base |
performance
| Key | Value |
|---|---|
| training_loss | 1.39 |
training
| Key | Value |
|---|---|
| global_batch_size_phase2 | 32768 |
| global_batch_size_phase1 | 65536 |
| iterations_phase1 | 7038 |
| LR_phase2 | 0.004 |
| LR_phase1 | 0.006 |
| iterations_phase2 | 1563 |
| training_precision | AMP |
| bs_phase2 | 8 |
| warmup_proportion_phase2 | 0.128 |
| bs_phase1 | 64 |
| warmup_proportion_phase1 | 0.2843 |
| iterations | 8601 |