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 | February 7, 2023 UTC | |
criterions.py | 1.04 KB | February 7, 2023 UTC | |
data_utils.py | 24.7 KB | February 7, 2023 UTC | |
Dockerfile | 1.04 KB | February 7, 2023 UTC | |
Dockerfile-triton | 1.7 KB | February 7, 2023 UTC | |
ema.py | 2.8 KB | February 7, 2023 UTC | |
gpu_affinity.py | 5.26 KB | February 7, 2023 UTC | |
inference.py | 11.09 KB | February 7, 2023 UTC | |
LICENSE | 11.09 KB | February 7, 2023 UTC | |
LICENSE AGREEMENT | 3.41 KB | February 7, 2023 UTC | |
log_helper.py | 7.17 KB | February 7, 2023 UTC | |
modeling.py | 15.97 KB | February 7, 2023 UTC | |
NOTICE | 314 B | February 7, 2023 UTC | |
README.md | 30.98 KB | February 7, 2023 UTC | |
requirements.txt | 88 B | February 7, 2023 UTC | |
TFT_architecture.PNG | 968.73 KB | February 7, 2023 UTC | |
tft_torchhub.py | 4.21 KB | February 7, 2023 UTC | |
train.py | 12.43 KB | February 7, 2023 UTC |
