diff --git a/examples/benchmarks/TFT/README.md b/examples/benchmarks/TFT/README.md new file mode 100644 index 000000000..03d45e087 --- /dev/null +++ b/examples/benchmarks/TFT/README.md @@ -0,0 +1,10 @@ +# Temporal Fusion Transformers Benchmark +## Source +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 diff --git a/examples/benchmarks/TFT/requirements.txt b/examples/benchmarks/TFT/requirements.txt new file mode 100644 index 000000000..04234aaed --- /dev/null +++ b/examples/benchmarks/TFT/requirements.txt @@ -0,0 +1,3 @@ +tensorflow-gpu==1.15.0 +numpy == 1.19.4 +pandas==1.1.0 \ No newline at end of file