ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
TFT PyTorch codebase
Resource
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
TFT PyTorch codebase

PyTorch codebase for training and using TFT model

NameSizeUpdatedActions
configuration.py
6.39 KBSeptember 22, 2022 UTC
criterions.py
1.04 KBSeptember 22, 2022 UTC
data_utils.py
24.7 KBSeptember 22, 2022 UTC
Dockerfile
1.04 KBSeptember 22, 2022 UTC
Dockerfile-triton
1.7 KBSeptember 22, 2022 UTC
ema.py
2.8 KBSeptember 22, 2022 UTC
gpu_affinity.py
5.26 KBSeptember 22, 2022 UTC
inference.py
11.09 KBSeptember 22, 2022 UTC
LICENSE
11.09 KBSeptember 22, 2022 UTC
LICENSE AGREEMENT
3.41 KBSeptember 22, 2022 UTC
log_helper.py
7.17 KBSeptember 22, 2022 UTC
modeling.py
15.97 KBSeptember 22, 2022 UTC
NOTICE
314 BSeptember 22, 2022 UTC
README.md
30.98 KBSeptember 22, 2022 UTC
requirements.txt
88 BSeptember 22, 2022 UTC
TFT_architecture.PNG
968.73 KBSeptember 22, 2022 UTC
train.py
12.43 KBSeptember 22, 2022 UTC