Model
TFT Base PyTorch checkpoint trained with AMP on Electricity dataset
Use the NGC CLI to download:
Copied!
1 Version
21.06.0Selected11/18/2021 10:11 PM UTC6.44 MBAccuracy: 040 EpochsBatch Size: 16384GPU: V100 Copied!
21.06.0Selected
11/18/2021 10:11 PM UTC6.44 MBAccuracy: 040 EpochsBatch Size: 16384GPU: V100
Copied!
architecture
| Key | Value |
|---|---|
| type | base |
performance
| Key | Value |
|---|---|
| P10 | 0.027 |
| P50 | 0.054 |
| P90 | 0.028 |
training
| Key | Value |
|---|---|
| batch_size | 16384 |
| bs_per_gpu | 2048 |
| clip_grad | 0.0 |
| dataset | Electricity |
| dropout | 0.1 |
| epochs | 40 |
| lr | 0.003 |
| training_precision | AMP |