NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
Transformer-XL for PyTorch
Resource
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
Transformer-XL for PyTorch

Transformer-XL is a transformer-based language model with a segment-level recurrence and a novel relative positional encoding.

Changelog

  • June 2020
    • Added support for NVIDIA DGX A100
    • Updated default NGC container to pytorch-20.06-py3
  • December 2019
    • Added support for the large Transformer-XL model trained on WikiText-103 dataset, the large model was trained on NVIDIA DGX-1, NVIDIA DGX-2 and on 8x NVIDIA DGX-2H (multi-node training)
    • Updated default NGC container to pytorch-19.11-py3
    • Added support for inference with TorchScript
  • October 2019
    • Initial release
    • Support for FP32 and mixed precision training on NVIDIA DGX-1, NVIDIA DGX-2, and inference on NVIDIA Tesla V100 16GB and NVIDIA T4

Known issues

There are no known issues with this model.