diff --git a/examples/benchmarks/TFT/README.md b/examples/benchmarks/TFT/README.md index 03d45e087..6d605a1bd 100644 --- a/examples/benchmarks/TFT/README.md +++ b/examples/benchmarks/TFT/README.md @@ -1,10 +1,12 @@ # Temporal Fusion Transformers Benchmark ## Source -https://github.com/google-research/google-research/tree/master/tft +**Reference**: Lim, Bryan, et al. "Temporal fusion transformers for interpretable multi-horizon time series forecasting." arXiv preprint arXiv:1912.09363 (2019). + +**GitHub**: https://github.com/google-research/google-research/tree/master/tft ## Run the Workflow Users can follow the ``workflow_by_code_tft.py`` to run the benchmark. ### Notes 1. The model must run in GPU, or an error will be raised. -2. New datasets should be registered in ``data_formatters``, for detail please visit the source. \ No newline at end of file +2. New datasets should be registered in ``data_formatters``, for detail please visit the source.