NGC Catalog
CLASSIC
Welcome Guest
Resources
TFT PyTorch codebase

TFT PyTorch codebase

For downloads and more information, please view on a desktop device.
Logo for TFT PyTorch codebase
Description
PyTorch codebase for training and using TFT model
Publisher
NVIDIA Deep Learning Examples
Latest Version
22.11.0
Modified
April 6, 2023
Compressed Size
4.22 MB

TFT PyTorch codebase

PyTorch codebase for training and using TFT model. Refer to the following guides for the use-cases of this codebase, setup instructions and performance numbers:

  • Temporal Fusion Transformer for PyTorch
  • Triton deployment of TFT PyTorch model