Model
BERT Distilled 6L-768D PyTorch Phase2 checkpoint pretrained using 67K steps on seqLen128 and 6k steps on seqLen512
Use the NGC CLI to download:
Copied!
1 Version
20.12.0_ampSelected07/07/2022 2:26 PM UTC253.44 MBAccuracy: 00 EpochsBatch Size: 0GPU: V100 Copied!
20.12.0_ampSelected
07/07/2022 2:26 PM UTC253.44 MBAccuracy: 00 EpochsBatch Size: 0GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| attn_heads | 12 |
| task | pretraining |
| hidden_size | 768 |
| di | 3072 |
| layers | 6 |
| type | Dist-6L-768D |
performance
| Key | Value |
|---|---|
| loss | 0.10294 |
training
| Key | Value |
|---|---|
| training_precision | AMP |
| dataset | Wikipedia+BookCorpus |