NVIDIA Deep Learning Examples
TFT PyTorch codebase
Resource
NVIDIA Deep Learning Examples
TFT PyTorch codebase

PyTorch codebase for training and using TFT model

NameSizeUpdatedActions
configuration.py
6.39 KBApril 6, 2023 UTC
criterions.py
1.34 KBApril 6, 2023 UTC
data_utils.py
27.03 KBApril 6, 2023 UTC
Dockerfile
1.04 KBApril 6, 2023 UTC
Dockerfile-triton
1.7 KBApril 6, 2023 UTC
ema.py
2.8 KBApril 6, 2023 UTC
gpu_affinity.py
5.26 KBApril 6, 2023 UTC
inference.py
11.41 KBApril 6, 2023 UTC
LICENSE
11.09 KBApril 6, 2023 UTC
LICENSE AGREEMENT
3.41 KBApril 6, 2023 UTC
log_helper.py
7.17 KBApril 6, 2023 UTC
modeling.py
20.29 KBApril 6, 2023 UTC
NOTICE
314 BApril 6, 2023 UTC
README.md
31.16 KBApril 6, 2023 UTC
requirements.txt
88 BApril 6, 2023 UTC
TFT_architecture.PNG
968.73 KBApril 6, 2023 UTC
tft_torchhub.py
4.21 KBApril 6, 2023 UTC
train.py
11.47 KBApril 6, 2023 UTC

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.