Use the NGC CLI to download:
Copied!
Resource
PyTorch codebase for training and using TFT model
Use the NGC CLI to download:
Copied!
| Name | Size | Updated | Actions |
|---|---|---|---|
configuration.py | 6.39 KB | April 6, 2023 UTC | |
criterions.py | 1.34 KB | April 6, 2023 UTC | |
data_utils.py | 27.03 KB | April 6, 2023 UTC | |
Dockerfile | 1.04 KB | April 6, 2023 UTC | |
Dockerfile-triton | 1.7 KB | April 6, 2023 UTC | |
ema.py | 2.8 KB | April 6, 2023 UTC | |
gpu_affinity.py | 5.26 KB | April 6, 2023 UTC | |
inference.py | 11.41 KB | April 6, 2023 UTC | |
LICENSE | 11.09 KB | April 6, 2023 UTC | |
LICENSE AGREEMENT | 3.41 KB | April 6, 2023 UTC | |
log_helper.py | 7.17 KB | April 6, 2023 UTC | |
modeling.py | 20.29 KB | April 6, 2023 UTC | |
NOTICE | 314 B | April 6, 2023 UTC | |
README.md | 31.16 KB | April 6, 2023 UTC | |
requirements.txt | 88 B | April 6, 2023 UTC | |
TFT_architecture.PNG | 968.73 KB | April 6, 2023 UTC | |
tft_torchhub.py | 4.21 KB | April 6, 2023 UTC | |
train.py | 11.47 KB | April 6, 2023 UTC |
