Model
Tranformer-XL Base PyTorch checkpoint trained with AMP
Use the NGC CLI to download:
Copied!
1 Version
19.11.0_ampSelected09/22/2022 5:11 PM UTC2.15 GBAccuracy: 00 EpochsBatch Size: 256GPU: V100 Copied!
19.11.0_ampSelected
09/22/2022 5:11 PM UTC2.15 GBAccuracy: 00 EpochsBatch Size: 256GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| type | Base |
performance
| Key | Value |
|---|---|
| perplexity | 23.19 |
training
| Key | Value |
|---|---|
| batch_size | 256 |
| learning rate | 0.01 |
| gpus | 8x V100 16G |
| training_precision | AMP |
| steps | 40000 |
| dataset | WikiText-103 |